Home
last modified time | relevance | path

Searched refs:mul_node (Results 1 – 8 of 8) 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.cc52 const NodeDef& mul_node = match.node; in FoldBatchNorms() local
63 {mul_node, conv_node, input_node, weights_node, in FoldBatchNorms()
112 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/lite/delegates/gpu/common/transformations/
Dfuse_mul_to_conv_test.cc64 auto mul_node = graph.NewNode(); in TEST() local
65 mul_node->operation.type = ToString(OperationType::MUL); in TEST()
66 mul_node->operation.attributes = mul_attr; in TEST()
71 ASSERT_TRUE(AddOutput(&graph, mul_node, &output).ok()); in TEST()
75 ASSERT_TRUE(ConnectTwoNodes(&graph, conv_node, mul_node, &link1).ok()); in TEST()
115 auto mul_node = graph.NewNode(); in TEST() local
116 mul_node->operation.type = ToString(OperationType::MUL); in TEST()
117 mul_node->operation.attributes = mul_attr; in TEST()
119 ASSERT_TRUE(graph.AddConsumer(mul_node->id, input->id).ok()); in TEST()
126 ASSERT_TRUE(ConnectTwoNodes(&graph, mul_node, conv_node, &link1).ok()); in TEST()
Dfuse_mul_to_conv.cc49 auto& mul_node = *sequence[1]; in ApplyToNodesSequence() local
50 if (mul_node.operation.type != ToString(OperationType::MUL) || in ApplyToNodesSequence()
51 !mul_node.operation.attributes.has_value()) { in ApplyToNodesSequence()
56 absl::any_cast<ElementwiseAttributes>(mul_node.operation.attributes); in ApplyToNodesSequence()
92 absl::Status status = RemoveFollowingNode(graph, &mul_node, &conv_node); in ApplyToNodesSequence()
113 auto& mul_node = *sequence[0]; in ApplyToNodesSequence() local
114 if (mul_node.operation.type != ToString(OperationType::MUL) || in ApplyToNodesSequence()
115 !mul_node.operation.attributes.has_value()) { in ApplyToNodesSequence()
120 absl::any_cast<ElementwiseAttributes>(mul_node.operation.attributes); in ApplyToNodesSequence()
156 absl::Status status = RemovePrecedingNode(graph, &mul_node, &conv_node); in ApplyToNodesSequence()
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dlayout_optimizer_test.cc464 auto mul_node = node_map_output.GetNode("mul"); in TEST_F() local
465 EXPECT_EQ(mul_node->input(0), "scalar_mul"); in TEST_F()
466 EXPECT_EQ(mul_node->input(1), "i5"); in TEST_F()
843 auto mul_node = node_map.GetNode("mul"); in TEST_F() local
844 EXPECT_EQ(mul_node->input(0), "scalar"); in TEST_F()
845 EXPECT_EQ(mul_node->input(1), "Conv2D"); in TEST_F()
860 auto mul_node = node_map.GetNode("mul"); in TEST_F() local
861 EXPECT_EQ(mul_node->input(0), "Conv2D"); in TEST_F()
862 EXPECT_EQ(mul_node->input(1), "scalar"); in TEST_F()
880 auto mul_node = node_map.GetNode("mul"); in TEST_F() local
[all …]
Darithmetic_optimizer_test.cc401 const NodeDef* mul_node = node_map.GetNode(HoistMulName("Add_6")); in TEST_F() local
402 ASSERT_NE(mul_node, nullptr); in TEST_F()
403 ASSERT_EQ(mul_node->input_size(), 2); in TEST_F()
404 EXPECT_EQ(mul_node->input(0), "Placeholder"); in TEST_F()
405 EXPECT_EQ(mul_node->input(1), HoistAddName("Add_6")); in TEST_F()
/external/tensorflow/tensorflow/core/grappler/
Dutils_test.cc357 const NodeDef* mul_node = node_map.GetNode("mul"); in TEST_F() local
367 EXPECT_EQ(NumControlInputs(*mul_node), 0); in TEST_F()
394 const NodeDef* mul_node = node_map.GetNode("mul"); in TEST_F() local
400 EXPECT_EQ(NumControlInputs(*mul_node), 0); in TEST_F()