Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/python/kernel_tests/
Dbatch_gather_op_test.py53 gather_t = array_ops.batch_gather(params, indices_tf)
69 gather_t = array_ops.batch_gather(params, indices_tf)
84 gather_t = array_ops.batch_gather(params, indices_tf)
97 self.evaluate(array_ops.batch_gather(params, indices_tf)))
103 gather_t = array_ops.batch_gather(params, indices)
110 self.evaluate(array_ops.batch_gather(params, [7]))
119 self.evaluate(array_ops.batch_gather(params, indices)),
/external/tensorflow/tensorflow/python/ops/ragged/
Dragged_batch_gather_ops.py35 def batch_gather(params, indices, name=None): function
68 return array_ops.batch_gather(params, indices, name)
91 batch_gather(params.values, indices.values), indices.row_splits)
Dragged_batch_gather_op_test.py151 result = ragged_batch_gather_ops.batch_gather(params, indices)
482 ragged_batch_gather_ops.batch_gather(params, indices)
486 ragged_batch_gather_ops.batch_gather(params, ragged_indices)
531 ragged_batch_gather_ops.batch_gather(params, indices)
Dragged_batch_gather_with_default_op.py143 return array_ops.batch_gather(
Dragged_dispatch.py429 (array_ops.batch_gather, ragged_batch_gather_ops.batch_gather,
Dragged_dispatch_test.py490 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.py3394 def batch_gather(params, indices, name=None): function
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.pbtxt876 name: "batch_gather"