| /external/tensorflow/tensorflow/tools/graph_transforms/ |
| remove_device_test.cc | 45 mul_node1->add_input("add_node2");
48 NodeDef* add_node2 = graph_def.add_node(); local
49 add_node2->set_name("add_node2");
50 add_node2->set_op("Add");
51 add_node2->add_input("const_node1");
52 add_node2->add_input("const_node2");
53 add_node2->set_device("//device:GPU:1");
76 add_node4->add_input("add_node2");
88 EXPECT_EQ("", node_lookup.at("add_node2")->device())
[all...] |
| remove_attribute_test.cc | 44 mul_node1->add_input("add_node2");
49 NodeDef* add_node2 = graph_def.add_node(); local
50 add_node2->set_name("add_node2");
51 add_node2->set_op("Add");
52 add_node2->add_input("const_node1");
53 add_node2->add_input("const_node2");
54 SetNodeAttr<int32>("foo", 46, add_node2);
55 SetNodeAttr<int32>("bob", 23, add_node2);
56 SetNodeAttr<string>("bar", "something else", add_node2);
[all...] |
| rename_attribute_test.cc | 44 mul_node1->add_input("add_node2");
49 NodeDef* add_node2 = graph_def.add_node(); local
50 add_node2->set_name("add_node2");
51 add_node2->set_op("Add");
52 add_node2->add_input("const_node1");
53 add_node2->add_input("const_node2");
54 AddNodeAttr("foo", 46, add_node2);
55 AddNodeAttr("bob", 23, add_node2);
56 AddNodeAttr("bar", "something else", add_node2);
[all...] |
| obfuscate_names_test.cc | 44 add_node1->add_input("add_node2");
47 NodeDef* add_node2 = graph_def.add_node(); local
48 add_node2->set_name("add_node2");
49 add_node2->set_op("Add");
50 add_node2->add_input("const_node1");
51 add_node2->add_input("const_node2");
83 EXPECT_EQ(0, node_lookup.count("add_node2"));
|
| rename_op_test.cc | 44 mul_node1->add_input("add_node2");
47 NodeDef* add_node2 = graph_def.add_node(); local
48 add_node2->set_name("add_node2");
49 add_node2->set_op("Add");
50 add_node2->add_input("const_node1");
51 add_node2->add_input("const_node2");
74 add_node4->add_input("add_node2");
91 EXPECT_EQ(1, node_lookup.count("add_node2"));
92 EXPECT_EQ("Add", node_lookup.at("add_node2")->op())
[all...] |
| sort_by_execution_order_test.cc | 105 add_node1->add_input("add_node2");
108 NodeDef* add_node2 = graph_def.add_node(); local
109 add_node2->set_name("add_node2");
110 add_node2->set_op("Add");
111 add_node2->add_input("const_node1");
112 add_node2->add_input("const_node2");
142 EXPECT_GT(6, order["add_node2"]);
156 add_node1->add_input("add_node2");
159 NodeDef* add_node2 = graph_def.add_node() local
[all...] |
| set_device_test.cc | 44 mul_node1->add_input("add_node2");
47 NodeDef* add_node2 = graph_def.add_node(); local
48 add_node2->set_name("add_node2");
49 add_node2->set_op("Add");
50 add_node2->add_input("const_node1");
51 add_node2->add_input("const_node2");
52 add_node2->set_device("/device:GPU:1");
75 add_node4->add_input("add_node2");
95 EXPECT_EQ("/device:CPU:0", node_lookup.at("add_node2")->device())
[all...] |
| insert_logging_test.cc | 57 Output add_node2 = local
58 Add(root.WithOpName("add_node2"), const_node1, const_node2);
61 Output mul_node1 = Mul(root.WithOpName("mul_node1"), add_node2, add_node3);
85 EXPECT_EQ(1, print_inputs.count("add_node2:0"));
104 Output add_node2 = local
105 Add(root.WithOpName("add_node2"), const_node1, const_node2);
108 Output mul_node1 = Mul(root.WithOpName("mul_node1"), add_node2, add_node3);
134 EXPECT_EQ(1, print_inputs.count("add_node2:0"));
153 Output add_node2 = local
154 Add(root.WithOpName("add_node2"), const_node1, const_node2)
[all...] |
| backports_test.cc | 48 mul_node1->add_input("add_node2");
51 NodeDef* add_node2 = graph_def.add_node(); local
52 add_node2->set_name("add_node2");
53 add_node2->set_op("Add");
54 add_node2->add_input("const_node1");
55 add_node2->add_input("const_node2");
|
| remove_nodes_test.cc | 44 add_node1->add_input("add_node2");
47 NodeDef* add_node2 = graph_def.add_node(); local
48 add_node2->set_name("add_node2");
49 add_node2->set_op("Add");
50 add_node2->add_input("identity_node1");
51 add_node2->add_input("identity_node2");
89 add_node4->add_input("add_node2");
103 EXPECT_EQ("add_node2", node_lookup.at("add_node1")->input(0));
105 EXPECT_EQ(1, node_lookup.count("add_node2"));
[all...] |
| strip_unused_nodes_test.cc | 76 add_node1->add_input("add_node2");
79 NodeDef* add_node2 = graph_def.add_node(); local
80 add_node2->set_name("add_node2");
81 add_node2->set_op("Add");
82 add_node2->add_input("const_node1");
83 add_node2->add_input("const_node2");
110 add_node4->add_input("add_node2");
120 EXPECT_EQ(1, node_lookup.count("add_node2"));
|
| transform_utils_test.cc | 709 *(mul_node->mutable_input()->Add()) = "add_node2:0";
719 NodeDef* add_node2 = graph_def.mutable_node()->Add(); local
720 add_node2->set_op("Add");
721 add_node2->set_name("add_node2");
722 *(add_node2->mutable_input()->Add()) = "missing_input3";
723 *(add_node2->mutable_input()->Add()) = "const_node1:0";
724 *(add_node2->mutable_input()->Add()) = "^const_node2";
739 (invalid_input.first == "add_node2"));
744 } else if (invalid_input.first == "add_node2") {
767 NodeDef* add_node2 = invalid_graph_def.mutable_node()->Add(); local
[all...] |
