| /external/tensorflow/tensorflow/tools/graph_transforms/ |
| remove_device_test.cc | 46 mul_node1->add_input("add_node3");
55 NodeDef* add_node3 = graph_def.add_node(); local
56 add_node3->set_name("add_node3");
57 add_node3->set_op("Add");
58 add_node3->add_input("const_node1");
59 add_node3->add_input("const_node3");
77 add_node4->add_input("add_node3");
|
| obfuscate_names_test.cc | 45 add_node1->add_input("add_node3");
53 NodeDef* add_node3 = graph_def.add_node(); local
54 add_node3->set_name("add_node3");
55 add_node3->set_op("Add");
56 add_node3->add_input("const_node3");
57 add_node3->add_input("const_node4");
84 EXPECT_EQ(0, node_lookup.count("add_node3"));
|
| rename_op_test.cc | 45 mul_node1->add_input("add_node3");
53 NodeDef* add_node3 = graph_def.add_node(); local
54 add_node3->set_name("add_node3");
55 add_node3->set_op("Add");
56 add_node3->add_input("const_node1");
57 add_node3->add_input("const_node3");
75 add_node4->add_input("add_node3");
93 EXPECT_EQ(1, node_lookup.count("add_node3"));
94 EXPECT_EQ("Add", node_lookup.at("add_node3")->op())
[all...] |
| sort_by_execution_order_test.cc | 106 add_node1->add_input("add_node3");
114 NodeDef* add_node3 = graph_def.add_node(); local
115 add_node3->set_name("add_node3");
116 add_node3->set_op("Add");
117 add_node3->add_input("const_node3");
118 add_node3->add_input("const_node4");
143 EXPECT_GT(6, order["add_node3"]);
157 add_node1->add_input("add_node3");
165 NodeDef* add_node3 = graph_def.add_node() local
[all...] |
| remove_attribute_test.cc | 45 mul_node1->add_input("add_node3");
58 NodeDef* add_node3 = graph_def.add_node(); local
59 add_node3->set_name("add_node3");
60 add_node3->set_op("Add");
61 add_node3->add_input("const_node1");
62 add_node3->add_input("const_node3");
80 add_node4->add_input("add_node3");
|
| rename_attribute_test.cc | 45 mul_node1->add_input("add_node3");
58 NodeDef* add_node3 = graph_def.add_node(); local
59 add_node3->set_name("add_node3");
60 add_node3->set_op("Add");
61 add_node3->add_input("const_node1");
62 add_node3->add_input("const_node3");
80 add_node4->add_input("add_node3");
|
| set_device_test.cc | 45 mul_node1->add_input("add_node3");
54 NodeDef* add_node3 = graph_def.add_node(); local
55 add_node3->set_name("add_node3");
56 add_node3->set_op("Add");
57 add_node3->add_input("const_node1");
58 add_node3->add_input("const_node3");
76 add_node4->add_input("add_node3");
96 EXPECT_EQ("/device:CPU:0", node_lookup.at("add_node3")->device());
119 EXPECT_EQ("/device:GPU:0", node_lookup.at("add_node3")->device())
[all...] |
| insert_logging_test.cc | 59 Output add_node3 = local
60 Add(root.WithOpName("add_node3"), const_node1, const_node3);
61 Output mul_node1 = Mul(root.WithOpName("mul_node1"), add_node2, add_node3);
86 EXPECT_EQ(1, print_inputs.count("add_node3:0"));
106 Output add_node3 = local
107 Add(root.WithOpName("add_node3"), const_node1, const_node3);
108 Output mul_node1 = Mul(root.WithOpName("mul_node1"), add_node2, add_node3);
135 EXPECT_EQ(1, print_inputs.count("add_node3:0"));
155 Output add_node3 = local
156 Add(root.WithOpName("add_node3"), const_node1, const_node3)
[all...] |
| backports_test.cc | 49 mul_node1->add_input("add_node3");
57 NodeDef* add_node3 = graph_def.add_node(); local
58 add_node3->set_name("add_node3");
59 add_node3->set_op("Add");
60 add_node3->add_input("const_node1");
61 add_node3->add_input("const_node3");
|
| remove_nodes_test.cc | 45 add_node1->add_input("add_node3");
53 NodeDef* add_node3 = graph_def.add_node(); local
54 add_node3->set_name("add_node3");
55 add_node3->set_op("Add");
56 add_node3->add_input("identity_node1");
57 add_node3->add_input("const_node3");
90 add_node4->add_input("add_node3");
104 EXPECT_EQ("add_node3", node_lookup.at("add_node1")->input(1));
108 EXPECT_EQ(1, node_lookup.count("add_node3"));
[all...] |
| strip_unused_nodes_test.cc | 77 add_node1->add_input("add_node3");
85 NodeDef* add_node3 = graph_def.add_node(); local
86 add_node3->set_name("add_node3");
87 add_node3->set_op("Add");
88 add_node3->add_input("const_node1");
89 add_node3->add_input("const_node3");
111 add_node4->add_input("add_node3");
121 EXPECT_EQ(1, node_lookup.count("add_node3"));
|
