| /external/tensorflow/tensorflow/tools/graph_transforms/ |
| remove_device_test.cc | 51 add_node2->add_input("const_node1");
58 add_node3->add_input("const_node1");
61 NodeDef* const_node1 = graph_def.add_node(); local
62 const_node1->set_name("const_node1");
63 const_node1->set_op("Const");
|
| remove_nodes_test.cc | 62 identity_node1->add_input("const_node1");
74 NodeDef* const_node1 = graph_def.add_node(); local
75 const_node1->set_name("const_node1");
76 const_node1->set_op("Const");
106 EXPECT_EQ("const_node1", node_lookup.at("add_node2")->input(0));
109 EXPECT_EQ("const_node1", node_lookup.at("add_node3")->input(0));
117 EXPECT_EQ(1, node_lookup.count("const_node1"));
118 EXPECT_EQ("Const", node_lookup.at("const_node1")->op());
128 NodeDef* const_node1 = graph_def.add_node() local
167 NodeDef* const_node1 = graph_def.add_node(); local
217 NodeDef* const_node1 = graph_def.add_node(); local
[all...] |
| insert_logging_test.cc | 51 Output const_node1 = local
52 Const(root.WithOpName("const_node1"), Input::Initializer(const_tensor));
58 Add(root.WithOpName("add_node2"), const_node1, const_node2);
60 Add(root.WithOpName("add_node3"), const_node1, const_node3);
88 EXPECT_EQ(1, print_inputs.count("const_node1:0"));
98 Output const_node1 = local
99 Const(root.WithOpName("const_node1"), Input::Initializer(const_tensor));
105 Add(root.WithOpName("add_node2"), const_node1, const_node2);
107 Add(root.WithOpName("add_node3"), const_node1, const_node3);
137 EXPECT_EQ(0, print_inputs.count("const_node1:0"))
147 Output const_node1 = local
[all...] |
| obfuscate_names_test.cc | 50 add_node2->add_input("const_node1");
59 NodeDef* const_node1 = graph_def.add_node(); local
60 const_node1->set_name("const_node1");
61 const_node1->set_op("Const");
77 ObfuscateNames(graph_def, {{"const_node1"}, {"add_node1"}}, &result));
85 EXPECT_EQ(1, node_lookup.count("const_node1"));
|
| rename_op_test.cc | 50 add_node2->add_input("const_node1");
56 add_node3->add_input("const_node1");
59 NodeDef* const_node1 = graph_def.add_node(); local
60 const_node1->set_name("const_node1");
61 const_node1->set_op("Const");
97 EXPECT_EQ(1, node_lookup.count("const_node1"));
98 EXPECT_EQ("Const", node_lookup.at("const_node1")->op());
|
| sort_by_execution_order_test.cc | 111 add_node2->add_input("const_node1");
120 NodeDef* const_node1 = graph_def.add_node(); local
121 const_node1->set_name("const_node1");
122 const_node1->set_op("Const");
144 EXPECT_GT(5, order["const_node1"]);
162 add_node2->add_input("const_node1");
168 add_node3->add_input("const_node1");
171 NodeDef* const_node1 = graph_def.add_node(); local
172 const_node1->set_name("const_node1")
[all...] |
| remove_attribute_test.cc | 52 add_node2->add_input("const_node1");
61 add_node3->add_input("const_node1");
64 NodeDef* const_node1 = graph_def.add_node(); local
65 const_node1->set_name("const_node1");
66 const_node1->set_op("Const");
|
| rename_attribute_test.cc | 52 add_node2->add_input("const_node1");
61 add_node3->add_input("const_node1");
64 NodeDef* const_node1 = graph_def.add_node(); local
65 const_node1->set_name("const_node1");
66 const_node1->set_op("Const");
|
| set_device_test.cc | 50 add_node2->add_input("const_node1");
57 add_node3->add_input("const_node1");
60 NodeDef* const_node1 = graph_def.add_node(); local
61 const_node1->set_name("const_node1");
62 const_node1->set_op("Const");
97 EXPECT_EQ("/device:CPU:0", node_lookup.at("const_node1")->device());
120 EXPECT_EQ("/device:GPU:0", node_lookup.at("const_node1")->device());
|
| backports_test.cc | 54 add_node2->add_input("const_node1");
60 add_node3->add_input("const_node1");
63 NodeDef* const_node1 = graph_def.add_node(); local
64 const_node1->set_name("const_node1");
65 const_node1->set_op("Const");
78 concat_node->add_input("const_node1");
95 EXPECT_EQ("const_node1", node_lookup.at("concat_node")->input(1));
97 EXPECT_EQ(1, node_lookup.count("const_node1"));
98 EXPECT_EQ("Const", node_lookup.at("const_node1")->op())
[all...] |
| strip_unused_nodes_test.cc | 82 add_node2->add_input("const_node1");
88 add_node3->add_input("const_node1");
91 NodeDef* const_node1 = graph_def.add_node(); local
92 const_node1->set_name("const_node1");
93 const_node1->set_op("Const");
123 EXPECT_EQ(1, node_lookup.count("const_node1"));
|
| transform_utils_test.cc | 710 *(mul_node->mutable_input()->Add()) = "^const_node1:0";
716 *(add_node1->mutable_input()->Add()) = "const_node1:0";
723 *(add_node2->mutable_input()->Add()) = "const_node1:0";
726 NodeDef* const_node1 = graph_def.mutable_node()->Add(); local
727 const_node1->set_op("Const");
728 const_node1->set_name("const_node1");
758 *(mul_node->mutable_input()->Add()) = "^const_node1:0";
764 *(add_node1->mutable_input()->Add()) = "const_node1:0";
771 *(add_node2->mutable_input()->Add()) = "const_node1:0"
774 NodeDef* const_node1 = invalid_graph_def.mutable_node()->Add(); local
[all...] |
