Home
last modified time | relevance | path

Searched refs:mul_node (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/tools/graph_transforms/
Dstrip_unused_nodes_test.cc138 NodeDef* mul_node = graph_def.add_node(); in TestSimplePlaceholder() local
139 mul_node->set_name("mul_node"); in TestSimplePlaceholder()
140 mul_node->set_op("Mul"); in TestSimplePlaceholder()
141 mul_node->add_input("b_node"); in TestSimplePlaceholder()
142 mul_node->add_input("c_node"); in TestSimplePlaceholder()
181 NodeDef* mul_node = graph_def.add_node(); in TestPlaceholderDefaultArgs() local
182 mul_node->set_name("mul_node"); in TestPlaceholderDefaultArgs()
183 mul_node->set_op("Mul"); in TestPlaceholderDefaultArgs()
184 mul_node->add_input("b_node"); in TestPlaceholderDefaultArgs()
185 mul_node->add_input("c_node"); in TestPlaceholderDefaultArgs()
[all …]
Dfold_batch_norms.cc53 const NodeDef& mul_node = match.node; in FoldBatchNorms() local
64 {mul_node, conv_node, input_node, weights_node, in FoldBatchNorms()
113 new_conv_node.set_name(mul_node.name()); in FoldBatchNorms()
Dtransform_utils_test.cc705 NodeDef* mul_node = graph_def.mutable_node()->Add(); in TestFindInvalidInputs() local
706 mul_node->set_op("Mul"); in TestFindInvalidInputs()
707 mul_node->set_name("mul_node"); in TestFindInvalidInputs()
708 *(mul_node->mutable_input()->Add()) = "add_node1"; in TestFindInvalidInputs()
709 *(mul_node->mutable_input()->Add()) = "add_node2:0"; in TestFindInvalidInputs()
710 *(mul_node->mutable_input()->Add()) = "^const_node1:0"; in TestFindInvalidInputs()
753 NodeDef* mul_node = invalid_graph_def.mutable_node()->Add(); in TestIsGraphValid() local
754 mul_node->set_op("Mul"); in TestIsGraphValid()
755 mul_node->set_name("mul_node"); in TestIsGraphValid()
756 *(mul_node->mutable_input()->Add()) = "add_node1"; in TestIsGraphValid()
[all …]
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dlayout_optimizer_test.cc444 auto mul_node = node_map_output.GetNode("mul"); in TEST_F() local
445 EXPECT_EQ(mul_node->input(0), "scalar_mul"); in TEST_F()
446 EXPECT_EQ(mul_node->input(1), "i5"); in TEST_F()
823 auto mul_node = node_map.GetNode("mul"); in TEST_F() local
824 EXPECT_EQ(mul_node->input(0), "scalar"); in TEST_F()
825 EXPECT_EQ(mul_node->input(1), "Conv2D"); in TEST_F()
840 auto mul_node = node_map.GetNode("mul"); in TEST_F() local
841 EXPECT_EQ(mul_node->input(0), "Conv2D"); in TEST_F()
842 EXPECT_EQ(mul_node->input(1), "scalar"); in TEST_F()
860 auto mul_node = node_map.GetNode("mul"); in TEST_F() local
[all …]
Darithmetic_optimizer_test.cc485 const NodeDef* mul_node = node_map.GetNode(HoistMulName("Add_6")); in TEST_F() local
486 ASSERT_NE(mul_node, nullptr); in TEST_F()
487 EXPECT_EQ(2, mul_node->input_size()); in TEST_F()
488 EXPECT_EQ("Placeholder", mul_node->input(0)); in TEST_F()
489 EXPECT_EQ(HoistAddName("Add_6"), mul_node->input(1)); in TEST_F()
/external/v8/src/compiler/arm64/
Dinstruction-selector-arm64.cc1151 Node* mul_node = madd_node.left().node(); in VisitWord32Sar() local
1154 Emit(kArm64Smull, smull_operand, g.UseRegister(mul_node->InputAt(0)), in VisitWord32Sar()
1155 g.UseRegister(mul_node->InputAt(1))); in VisitWord32Sar()