Searched defs:gather_v2 (Results 1 – 2 of 2) 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()
73 void TestSinglePartition(bool gather_v2, bool include_shared_init, in TestSinglePartition()
285 void TestMultiPartition(bool gather_v2, bool include_shared_init, in TestMultiPartition()
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | array_ops.py | 3380 def gather_v2(params, indices, validate_indices=None, axis=None, function
|