Searched refs:add_node1 (Results 1 - 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/tools/graph_transforms/
H A Dobfuscate_names_test.cc41 NodeDef* add_node1 = graph_def.add_node(); local
42 add_node1->set_name("add_node1");
43 add_node1->set_op("Add");
44 add_node1->add_input("add_node2");
45 add_node1->add_input("add_node3");
77 ObfuscateNames(graph_def, {{"const_node1"}, {"add_node1"}}, &result));
82 EXPECT_EQ(1, node_lookup.count("add_node1"));
H A Dsort_by_execution_order_test.cc102 NodeDef* add_node1 = graph_def.add_node(); local
103 add_node1->set_name("add_node1");
104 add_node1->set_op("Add");
105 add_node1->add_input("add_node2");
106 add_node1->add_input("add_node3");
141 EXPECT_EQ(6, order["add_node1"]);
153 NodeDef* add_node1 = graph_def.add_node(); local
154 add_node1->set_name("add_node1");
[all...]
H A Dremove_nodes_test.cc41 NodeDef* add_node1 = graph_def.add_node(); local
42 add_node1->set_name("add_node1");
43 add_node1->set_op("Add");
44 add_node1->add_input("add_node2");
45 add_node1->add_input("add_node3");
95 context.output_names = {"add_node1"};
102 EXPECT_EQ(1, node_lookup.count("add_node1"));
103 EXPECT_EQ("add_node2", node_lookup.at("add_node1")->input(0));
104 EXPECT_EQ("add_node3", node_lookup.at("add_node1")
[all...]
H A Dstrip_unused_nodes_test.cc73 NodeDef* add_node1 = graph_def.add_node(); local
74 add_node1->set_name("add_node1");
75 add_node1->set_op("Add");
76 add_node1->add_input("add_node2");
77 add_node1->add_input("add_node3");
115 graph_def, {{"dangling_input"}, {"add_node1"}}, &result));
119 EXPECT_EQ(1, node_lookup.count("add_node1"));
H A Dtransform_utils_test.cc708 *(mul_node->mutable_input()->Add()) = "add_node1";
712 NodeDef* add_node1 = graph_def.mutable_node()->Add(); local
713 add_node1->set_op("Add");
714 add_node1->set_name("add_node1");
715 *(add_node1->mutable_input()->Add()) = "missing_input1";
716 *(add_node1->mutable_input()->Add()) = "const_node1:0";
717 *(add_node1->mutable_input()->Add()) = "missing_input2";
738 EXPECT_TRUE((invalid_input.first == "add_node1") ||
740 if (invalid_input.first == "add_node1") {
760 NodeDef* add_node1 = invalid_graph_def.mutable_node()->Add(); local
[all...]

Completed in 110 milliseconds