Searched refs:MakeGather (Results 1 – 1 of 1) 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() function in tensorflow::graph_transforms::SparsifyGatherTest
137 MakeGather("gather", gather_v2, identity_node, input_node, &graph_def); in TestSinglePartition()
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()
|