Searched refs:w_node (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/tools/graph_transforms/ |
| D | sparsify_gather_test.cc | 82 NodeDef* w_node; in TestSinglePartition() local
92 w_node = CreateNode("w/part_1", "Const", {}, &graph_def); in TestSinglePartition()
93 SetNodeTensorAttr<float>("value", weights, w_node); in TestSinglePartition()
95 w_node = CreateNode("w/part_1", "VariableV2", {}, &graph_def); in TestSinglePartition()
104 {w_node, zeros_node}, &graph_def); in TestSinglePartition()
127 CreateNode("save/Assign", "Assign", {w_node, restore_node}, &graph_def); in TestSinglePartition()
133 SetNodeAttr("dtype", DT_FLOAT, w_node); in TestSinglePartition()
136 CreateNode("w/read", "Identity", {w_node}, &graph_def); in TestSinglePartition()
|