| /external/tensorflow/tensorflow/tools/graph_transforms/ |
| strip_unused_nodes_test.cc | 135 add_node->add_input("mul_node");
138 NodeDef* mul_node = graph_def.add_node(); local
139 mul_node->set_name("mul_node");
140 mul_node->set_op("Mul");
141 mul_node->add_input("b_node");
142 mul_node->add_input("c_node");
158 StripUnusedNodes(graph_def, {{"mul_node"}, {"add_node"}}, &result));
163 EXPECT_EQ(1, node_lookup.count("mul_node"));
164 EXPECT_EQ("Placeholder", node_lookup["mul_node"]->op())
181 NodeDef* mul_node = graph_def.add_node(); local
227 NodeDef* mul_node = graph_def.add_node(); local
[all...] |
| fold_batch_norms.cc | 38 {"Mul", // mul_node
53 const NodeDef& mul_node = match.node;
64 {mul_node, conv_node, input_node, weights_node,
113 new_conv_node.set_name(mul_node.name());
|
| transform_utils_test.cc | 705 NodeDef* mul_node = graph_def.mutable_node()->Add(); local
706 mul_node->set_op("Mul");
707 mul_node->set_name("mul_node");
708 *(mul_node->mutable_input()->Add()) = "add_node1";
709 *(mul_node->mutable_input()->Add()) = "add_node2:0";
710 *(mul_node->mutable_input()->Add()) = "^const_node1:0";
753 NodeDef* mul_node = invalid_graph_def.mutable_node()->Add(); local
754 mul_node->set_op("Mul");
755 mul_node->set_name("mul_node")
[all...] |
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| layout_optimizer_test.cc | 444 auto mul_node = node_map_output.GetNode("mul"); local
445 EXPECT_EQ(mul_node->input(0), "scalar_mul");
446 EXPECT_EQ(mul_node->input(1), "i5");
823 auto mul_node = node_map.GetNode("mul"); local
840 auto mul_node = node_map.GetNode("mul"); local
860 auto mul_node = node_map.GetNode("mul"); local
880 auto mul_node = node_map.GetNode("mul"); local
897 auto mul_node = node_map.GetNode("mul"); local
921 auto mul_node = node_map.GetNode("mul"); local
1214 auto mul_node = node_map.GetNode("mul"); local
[all...] |
| arithmetic_optimizer_test.cc | 485 const NodeDef* mul_node = node_map.GetNode(HoistMulName("Add_6")); local
486 ASSERT_NE(mul_node, nullptr);
487 EXPECT_EQ(2, mul_node->input_size());
488 EXPECT_EQ("Placeholder", mul_node->input(0));
489 EXPECT_EQ(HoistAddName("Add_6"), mul_node->input(1));
[all...] |
| /external/v8/src/compiler/arm64/ |
| instruction-selector-arm64.cc | 1151 Node* mul_node = madd_node.left().node(); local
[all...] |
