Searched refs:batch_gather (Results 1 – 11 of 11) sorted by relevance
| /external/tensorflow/tensorflow/python/kernel_tests/array_ops/ |
| D | batch_gather_op_test.py | 54 gather_t = array_ops.batch_gather(params, indices_tf)
70 gather_t = array_ops.batch_gather(params, indices_tf)
85 gather_t = array_ops.batch_gather(params, indices_tf)
98 self.evaluate(array_ops.batch_gather(params, indices_tf)))
105 gather_t = array_ops.batch_gather(params, indices)
113 self.evaluate(array_ops.batch_gather(params, [7]))
122 self.evaluate(array_ops.batch_gather(params, indices)),
|
| /external/tensorflow/tensorflow/python/ops/ragged/ |
| D | ragged_batch_gather_op_test.py | 150 result = ragged_batch_gather_ops.batch_gather(params, indices)
482 ragged_batch_gather_ops.batch_gather(params, indices)
487 ragged_batch_gather_ops.batch_gather(params, ragged_indices)
533 ragged_batch_gather_ops.batch_gather(params, indices)
|
| D | ragged_batch_gather_ops.py | 27 def batch_gather(params, indices, name=None): function
|
| D | ragged_batch_gather_with_default_op.py | 140 return array_ops.batch_gather(
|
| D | ragged_dispatch.py | 482 (array_ops.batch_gather, ragged_batch_gather_ops.batch_gather,
|
| D | ragged_dispatch_test.py | 396 op=array_ops.batch_gather,
|
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_GatherV2.pbtxt | 58 See also `tf.batch_gather` and `tf.gather_nd`.
|
| D | api_def_GatherNd.pbtxt | 127 See also `tf.gather` and `tf.batch_gather`.
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | array_ops.py | 5060 def batch_gather(params, indices, name=None): function
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.pbtxt | 912 name: "batch_gather"
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_generated_ops.td | 4959 See also `tf.gather` and `tf.batch_gather`.
5008 See also `tf.batch_gather` and `tf.gather_nd`.
|