Home
last modified time | relevance | path

Searched refs:batch_gather (Results 1 – 10 of 10) sorted by relevance

/external/tensorflow/tensorflow/python/kernel_tests/array_ops/
Dbatch_gather_op_test.py50 gather_t = array_ops.batch_gather(params, indices_tf)
66 gather_t = array_ops.batch_gather(params, indices_tf)
81 gather_t = array_ops.batch_gather(params, indices_tf)
94 self.evaluate(array_ops.batch_gather(params, indices_tf)))
101 gather_t = array_ops.batch_gather(params, indices)
109 self.evaluate(array_ops.batch_gather(params, [7]))
118 self.evaluate(array_ops.batch_gather(params, indices)),
/external/tensorflow/tensorflow/python/ops/ragged/
Dragged_batch_gather_ops.py26 @dispatch.dispatch_for_api(array_ops.batch_gather)
27 def batch_gather(params: ragged_tensor.RaggedOrDense, function
Dragged_batch_gather_op_test.py146 result = ragged_batch_gather_ops.batch_gather(params, indices)
478 ragged_batch_gather_ops.batch_gather(params, indices)
483 ragged_batch_gather_ops.batch_gather(params, ragged_indices)
529 ragged_batch_gather_ops.batch_gather(params, indices)
Dragged_batch_gather_with_default_op.py136 return array_ops.batch_gather(
Dragged_dispatch_test.py706 op=array_ops.batch_gather,
/external/tensorflow/tensorflow/core/api_def/base_api/
Dapi_def_GatherV2.pbtxt58 See also `tf.batch_gather` and `tf.gather_nd`.
Dapi_def_GatherNd.pbtxt127 See also `tf.gather` and `tf.batch_gather`.
/external/tensorflow/tensorflow/python/ops/
Darray_ops.py5336 def batch_gather(params, indices, name=None): function
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.pbtxt916 name: "batch_gather"
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/
Dtf_generated_ops.td5692 See also `tf.gather` and `tf.batch_gather`.
5741 See also `tf.batch_gather` and `tf.gather_nd`.