| /external/tensorflow/tensorflow/tools/graph_transforms/ |
| sparsify_gather_test.cc | 180 std::map<string, const NodeDef*> node_lookup; local
181 MapNamesToNodes(result, &node_lookup);
185 node_lookup.count("w/part_1/Initializer/zeros/shape_as_tensor"));
186 EXPECT_EQ(0, node_lookup.count("w/part_1/Initializer/zeros/Const"));
187 EXPECT_EQ(0, node_lookup.count("w/part_1/Initializer/zeros"));
188 EXPECT_EQ(0, node_lookup.count("w/part_1/Assign"));
190 EXPECT_EQ(1, node_lookup.count("ids"));
191 EXPECT_EQ("Const", node_lookup.at("ids")->op());
193 EXPECT_EQ(1, node_lookup.count("concat/node"));
196 EXPECT_EQ(0, node_lookup.count("linear/concat/axis"))
412 std::map<string, const NodeDef*> node_lookup; local
[all...] |
| obfuscate_names_test.cc | 79 std::map<string, const NodeDef*> node_lookup; local
80 MapNamesToNodes(result, &node_lookup);
82 EXPECT_EQ(1, node_lookup.count("add_node1"));
83 EXPECT_EQ(0, node_lookup.count("add_node2"));
84 EXPECT_EQ(0, node_lookup.count("add_node3"));
85 EXPECT_EQ(1, node_lookup.count("const_node1"));
86 EXPECT_EQ(0, node_lookup.count("const_node2"));
87 EXPECT_EQ(0, node_lookup.count("const_node3"));
88 EXPECT_EQ(0, node_lookup.count("const_node4"));
103 std::map<string, const NodeDef*> node_lookup; local
121 std::map<string, const NodeDef*> node_lookup; local
[all...] |
| strip_unused_nodes_test.cc | 62 std::map<string, const NodeDef*> node_lookup; local
63 MapNamesToNodes(result, &node_lookup);
64 EXPECT_EQ(1, node_lookup.count("add_node"));
65 EXPECT_EQ(1, node_lookup.count("a_node"));
66 EXPECT_EQ(1, node_lookup.count("b_node"));
67 EXPECT_EQ(0, node_lookup.count("c_node"));
117 std::map<string, const NodeDef*> node_lookup; local
118 MapNamesToNodes(result, &node_lookup);
119 EXPECT_EQ(1, node_lookup.count("add_node1"));
120 EXPECT_EQ(1, node_lookup.count("add_node2"))
160 std::map<string, const NodeDef*> node_lookup; local
206 std::map<string, const NodeDef*> node_lookup; local
254 std::map<string, const NodeDef*> node_lookup; local
[all...] |
| rename_op_test.cc | 87 std::map<string, const NodeDef*> node_lookup; local
88 MapNamesToNodes(result, &node_lookup);
89 EXPECT_EQ(1, node_lookup.count("mul_node1"));
90 EXPECT_EQ("Multiply", node_lookup.at("mul_node1")->op());
91 EXPECT_EQ(1, node_lookup.count("add_node2"));
92 EXPECT_EQ("Add", node_lookup.at("add_node2")->op());
93 EXPECT_EQ(1, node_lookup.count("add_node3"));
94 EXPECT_EQ("Add", node_lookup.at("add_node3")->op());
95 EXPECT_EQ(1, node_lookup.count("add_node4"));
96 EXPECT_EQ("Add", node_lookup.at("add_node4")->op())
[all...] |
| remove_nodes_test.cc | 100 std::map<string, const NodeDef*> node_lookup; local
101 MapNamesToNodes(result, &node_lookup);
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")->input(1));
105 EXPECT_EQ(1, node_lookup.count("add_node2"));
106 EXPECT_EQ("const_node1", node_lookup.at("add_node2")->input(0));
107 EXPECT_EQ("const_node2", node_lookup.at("add_node2")->input(1));
108 EXPECT_EQ(1, node_lookup.count("add_node3"));
109 EXPECT_EQ("const_node1", node_lookup.at("add_node3")->input(0))
155 std::map<string, const NodeDef*> node_lookup; local
204 std::map<string, const NodeDef*> node_lookup; local
256 std::map<string, const NodeDef*> node_lookup; local
[all...] |
| set_device_test.cc | 92 std::map<string, const NodeDef*> node_lookup; local
93 MapNamesToNodes(result, &node_lookup);
94 EXPECT_EQ("/device:CPU:0", node_lookup.at("mul_node1")->device());
95 EXPECT_EQ("/device:CPU:0", node_lookup.at("add_node2")->device());
96 EXPECT_EQ("/device:CPU:0", node_lookup.at("add_node3")->device());
97 EXPECT_EQ("/device:CPU:0", node_lookup.at("const_node1")->device());
98 EXPECT_EQ("/device:CPU:0", node_lookup.at("const_node2")->device());
99 EXPECT_EQ("/device:CPU:0", node_lookup.at("const_node3")->device());
100 EXPECT_EQ("/device:CPU:0", node_lookup.at("add_node4")->device());
115 std::map<string, const NodeDef*> node_lookup; local
[all...] |
| rename_attribute_test.cc | 94 std::map<string, const NodeDef*> node_lookup; local
95 MapNamesToNodes(wildcard_result, &node_lookup);
96 EXPECT_EQ(0, node_lookup.at("mul_node1")->attr().count("foo"));
97 EXPECT_EQ(1, node_lookup.at("mul_node1")->attr().count("baz"));
98 EXPECT_EQ(1, node_lookup.at("mul_node1")->attr().count("bar"));
99 EXPECT_EQ(0, node_lookup.at("add_node2")->attr().count("foo"));
100 EXPECT_EQ(1, node_lookup.at("add_node2")->attr().count("baz"));
101 EXPECT_EQ(1, node_lookup.at("add_node2")->attr().count("bar"));
102 EXPECT_EQ(1, node_lookup.at("add_node2")->attr().count("bob"));
117 MapNamesToNodes(targeted_result, &node_lookup);
[all...] |
| backports_test.cc | 89 std::map<string, const NodeDef*> node_lookup; local
90 MapNamesToNodes(result, &node_lookup);
91 EXPECT_EQ(1, node_lookup.count("concat_node"));
92 EXPECT_EQ("Concat", node_lookup.at("concat_node")->op());
93 EXPECT_EQ(0, node_lookup.at("concat_node")->attr().count("Tidx"));
94 EXPECT_EQ("const_node3", node_lookup.at("concat_node")->input(0));
95 EXPECT_EQ("const_node1", node_lookup.at("concat_node")->input(1));
96 EXPECT_EQ("const_node2", node_lookup.at("concat_node")->input(2));
97 EXPECT_EQ(1, node_lookup.count("const_node1"));
98 EXPECT_EQ("Const", node_lookup.at("const_node1")->op())
161 std::map<string, const NodeDef*> node_lookup; local
192 std::map<string, const NodeDef*> node_lookup; local
[all...] |
| remove_attribute_test.cc | 92 std::map<string, const NodeDef*> node_lookup; local
93 MapNamesToNodes(wildcard_result, &node_lookup);
94 EXPECT_EQ(0, node_lookup.at("mul_node1")->attr().count("foo"));
95 EXPECT_EQ(1, node_lookup.at("mul_node1")->attr().count("bar"));
96 EXPECT_EQ(0, node_lookup.at("add_node2")->attr().count("foo"));
97 EXPECT_EQ(1, node_lookup.at("add_node2")->attr().count("bar"));
98 EXPECT_EQ(1, node_lookup.at("add_node2")->attr().count("bob"));
111 MapNamesToNodes(targeted_result, &node_lookup);
112 EXPECT_EQ(0, node_lookup.at("mul_node1")->attr().count("foo"));
113 EXPECT_EQ(1, node_lookup.at("mul_node1")->attr().count("bar"))
[all...] |
| remove_device_test.cc | 85 std::map<string, const NodeDef*> node_lookup; local
86 MapNamesToNodes(result, &node_lookup);
87 EXPECT_EQ("", node_lookup.at("mul_node1")->device());
88 EXPECT_EQ("", node_lookup.at("add_node2")->device());
|
| round_weights_test.cc | 81 std::map<string, const NodeDef*> node_lookup; local
82 MapNamesToNodes(rounded_graph_def, &node_lookup);
83 EXPECT_EQ(1, node_lookup.count("input_op"));
84 const NodeDef* r_input_op = node_lookup.at("input_op");
86 EXPECT_EQ(1, node_lookup.count("weights_op"));
87 const NodeDef* r_weights_op = node_lookup.at("weights_op");
|
| quantize_weights_test.cc | 81 std::map<string, const NodeDef*> node_lookup; local
82 MapNamesToNodes(quantized_graph_def, &node_lookup);
83 EXPECT_EQ(1, node_lookup.count("input_op"));
84 const NodeDef* q_input_op = node_lookup.at("input_op");
86 EXPECT_EQ(1, node_lookup.count("weights_op"));
87 const NodeDef* q_weights_op = node_lookup.at("weights_op");
91 EXPECT_EQ(1, node_lookup.count(weights_const_name));
92 const NodeDef* q_weights_const = node_lookup.at(weights_const_name);
135 std::map<string, const NodeDef*> node_lookup; local
136 MapNamesToNodes(quantized_graph_def, &node_lookup);
[all...] |
| strip_unused_nodes.cc | 130 std::map<string, const NodeDef*> node_lookup; local
131 MapNamesToNodes(input_graph_def, &node_lookup);
145 if (!node_lookup.count(current_input)) {
149 const NodeDef* current_node = node_lookup[current_input];
|
