Searched refs:w_node2 (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/tools/graph_transforms/ |
| D | sparsify_gather_test.cc | 299 NodeDef* w_node2; in TestMultiPartition() local
313 w_node2 = CreateNode("w2/part_1", "Const", {}, &graph_def); in TestMultiPartition()
315 SetNodeTensorAttr<float>("value", weights, w_node2); in TestMultiPartition()
353 w_node2 = CreateNode("w2/part_1", "VariableV2", {}, &graph_def); in TestMultiPartition()
361 {w_node2, zeros_node2}, &graph_def); in TestMultiPartition()
363 CreateNode("save/Assign_1", "Assign", {w_node2, restore_node}, in TestMultiPartition()
372 SetNodeAttr("dtype", DT_FLOAT, w_node2); in TestMultiPartition()
377 CreateNode("w2/part_1/read", "Identity", {w_node2}, &graph_def); in TestMultiPartition()
|