Searched refs:restore_node (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/tools/graph_transforms/ |
| D | sparsify_gather_test.cc | 123 NodeDef* restore_node = CreateNode( in TestSinglePartition() local
127 CreateNode("save/Assign", "Assign", {w_node, restore_node}, &graph_def); in TestSinglePartition()
335 NodeDef* restore_node = CreateNode( in TestMultiPartition() local
351 CreateNode("save/Assign", "Assign", {w_node1, restore_node}, &graph_def); in TestMultiPartition()
363 CreateNode("save/Assign_1", "Assign", {w_node2, restore_node}, in TestMultiPartition()
|