Searched refs:gather_v2 (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/tools/graph_transforms/ |
| D | sparsify_gather_test.cc | 59 void MakeGather(StringPiece name, bool gather_v2, NodeDef* params, in MakeGather() argument
61 if (gather_v2) { in MakeGather()
73 void TestSinglePartition(bool gather_v2, bool include_shared_init, in TestSinglePartition() argument
137 MakeGather("gather", gather_v2, identity_node, input_node, &graph_def); in TestSinglePartition()
285 void TestMultiPartition(bool gather_v2, bool include_shared_init, in TestMultiPartition() argument
378 MakeGather("gather1", gather_v2, identity_node1, input_node, &graph_def); in TestMultiPartition()
379 MakeGather("gather2", gather_v2, identity_node2, input_node, &graph_def); in TestMultiPartition()
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | array_ops.py | 3369 return gen_array_ops.gather_v2(params, indices, axis, name=name)
3375 return gen_array_ops.gather_v2(params, indices, axis, name=name)
3380 def gather_v2(params, indices, validate_indices=None, axis=None, function
3386 gather.__doc__ = gather_v2.__doc__ = gen_array_ops.gather_v2.__doc__
|
| /external/tensorflow/tensorflow/python/ops/ragged/ |
| D | ragged_dispatch.py | 435 (array_ops.gather_v2, ragged_gather_ops.gather, ['params', 'indices']),
|
| D | ragged_dispatch_test.py | 527 op=array_ops.gather_v2,
|