Searched refs:_batch_gather (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/python/ops/ |
| D | array_ops.py | 3362 return _batch_gather(params, indices, batch_dims, axis)
3402 return _batch_gather(params, indices, batch_dims=indices.shape.ndims - 1)
3405 def _batch_gather(params, indices, batch_dims, axis=None): function
3479 result = _batch_gather(params, indices, batch_dims=batch_dims)
|