MCPcopy Create free account
hub / github.com/alibaba/nann / _BatchGatherGrad

Function _BatchGatherGrad

tensorflow/tensorflow/python/ops/array_grad.py:502–533  ·  view source on GitHub ↗

Returns the gradient of GatherV2 with batch dimensions.

(
    params_shape, values, indices, batch_dims, gather_dim_size)

Source from the content-addressed store, hash-verified

source not stored for this graph (policy: none)

Callers 1

_GatherV2GradFunction · 0.85

Calls 6

_GetBatchIndicesFunction · 0.85
reshapeMethod · 0.80
expand_dimsMethod · 0.45
sizeMethod · 0.45
shapeMethod · 0.45
concatMethod · 0.45

Tested by

no test coverage detected