| /external/tensorflow/tensorflow/tools/graph_transforms/ |
| remove_device_test.cc | 73 NodeDef* add_node4 = graph_def.add_node(); local
74 add_node4->set_name("add_node4");
75 add_node4->set_op("Add");
76 add_node4->add_input("add_node2");
77 add_node4->add_input("add_node3");
|
| rename_op_test.cc | 71 NodeDef* add_node4 = graph_def.add_node(); local
72 add_node4->set_name("add_node4");
73 add_node4->set_op("Add");
74 add_node4->add_input("add_node2");
75 add_node4->add_input("add_node3");
95 EXPECT_EQ(1, node_lookup.count("add_node4"));
96 EXPECT_EQ("Add", node_lookup.at("add_node4")->op());
|
| remove_attribute_test.cc | 76 NodeDef* add_node4 = graph_def.add_node(); local
77 add_node4->set_name("add_node4");
78 add_node4->set_op("Add");
79 add_node4->add_input("add_node2");
80 add_node4->add_input("add_node3");
|
| rename_attribute_test.cc | 76 NodeDef* add_node4 = graph_def.add_node(); local
77 add_node4->set_name("add_node4");
78 add_node4->set_op("Add");
79 add_node4->add_input("add_node2");
80 add_node4->add_input("add_node3");
|
| set_device_test.cc | 72 NodeDef* add_node4 = graph_def.add_node(); local
73 add_node4->set_name("add_node4");
74 add_node4->set_op("Add");
75 add_node4->add_input("add_node2");
76 add_node4->add_input("add_node3");
100 EXPECT_EQ("/device:CPU:0", node_lookup.at("add_node4")->device());
123 EXPECT_EQ("/device:GPU:0", node_lookup.at("add_node4")->device());
|
| remove_nodes_test.cc | 86 NodeDef* add_node4 = graph_def.add_node(); local
87 add_node4->set_name("add_node4");
88 add_node4->set_op("Add");
89 add_node4->add_input("add_node2");
90 add_node4->add_input("add_node3");
111 EXPECT_EQ(1, node_lookup.count("add_node4"));
112 EXPECT_EQ("add_node2", node_lookup.at("add_node4")->input(0));
113 EXPECT_EQ("add_node3", node_lookup.at("add_node4")->input(1));
|
| strip_unused_nodes_test.cc | 107 NodeDef* add_node4 = graph_def.add_node(); local
108 add_node4->set_name("add_node4");
109 add_node4->set_op("Add");
110 add_node4->add_input("add_node2");
111 add_node4->add_input("add_node3");
122 EXPECT_EQ(0, node_lookup.count("add_node4"));
|
| insert_logging_test.cc | 62 Output add_node4 = local
63 Add(root.WithOpName("add_node4"), mul_node1, const_node3);
66 CheckGraphCanRun(graph_def, {"add_node4"});
71 context.output_names = {"add_node4"};
74 CheckGraphCanRun(result, {"add_node4"});
87 EXPECT_EQ(0, print_inputs.count("add_node4:0"));
109 Output add_node4 = local
110 Add(root.WithOpName("add_node4"), mul_node1, const_node3);
113 CheckGraphCanRun(graph_def, {"add_node4"});
118 context.output_names = {"add_node4"};
158 Output add_node4 = local
[all...] |
