Searched refs:_batch_gather (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/python/ops/ragged/ |
| D | ragged_gather_ops.py | 138 return _batch_gather(params, indices, axis, batch_dims)
176 def _batch_gather(params, indices, axis, batch_dims): function
237 return _batch_gather(params, adjusted_indices, axis, batch_dims + 1)
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | array_ops.py | 5068 return _batch_gather(params, indices, batch_dims=indices.shape.ndims - 1)
5071 def _batch_gather(params, indices, batch_dims, axis=None): function
5145 result = _batch_gather(params, indices, batch_dims=batch_dims)
|