Searched refs:w_node1 (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/tools/graph_transforms/ |
| D | sparsify_gather_test.cc | 298 NodeDef* w_node1; in TestMultiPartition() local
312 w_node1 = CreateNode("w1/part_1", "Const", {}, &graph_def); in TestMultiPartition()
314 SetNodeTensorAttr<float>("value", weights, w_node1); in TestMultiPartition()
340 w_node1 = CreateNode("w1/part_1", "VariableV2", {}, &graph_def); in TestMultiPartition()
349 {w_node1, zeros_node1}, &graph_def); in TestMultiPartition()
351 CreateNode("save/Assign", "Assign", {w_node1, restore_node}, &graph_def); in TestMultiPartition()
371 SetNodeAttr("dtype", DT_FLOAT, w_node1); in TestMultiPartition()
375 CreateNode("w1/part_1/read", "Identity", {w_node1}, &graph_def); in TestMultiPartition()
|