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

Function _GetBatchIndices

tensorflow/tensorflow/python/ops/array_grad.py:481–499  ·  view source on GitHub ↗

Addds the batch offsets to the given indices and returns the results.

(params_shape, indices, batch_dims)

Source from the content-addressed store, hash-verified

source not stored for this graph (policy: none)

Callers 1

_BatchGatherGradFunction · 0.85

Calls 6

onesMethod · 0.80
reshapeMethod · 0.80
rangeFunction · 0.70
castMethod · 0.45
rangeMethod · 0.45
stackMethod · 0.45

Tested by

no test coverage detected