Home
last modified time | relevance | path

Searched refs:bias_add_node (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/tools/graph_transforms/
Dfold_old_batch_norms.cc162 NodeDef bias_add_node; in FuseScaleOffsetToConvWeights() local
163 bias_add_node.set_op("BiasAdd"); in FuseScaleOffsetToConvWeights()
164 bias_add_node.set_name(conv_output_name); in FuseScaleOffsetToConvWeights()
166 CopyNodeAttr(conv_node, "data_format", "data_format", &bias_add_node); in FuseScaleOffsetToConvWeights()
168 CopyNodeAttr(conv_node, "T", "T", &bias_add_node); in FuseScaleOffsetToConvWeights()
169 AddNodeInput(conv_node.name(), &bias_add_node); in FuseScaleOffsetToConvWeights()
170 AddNodeInput(bias_offset_node.name(), &bias_add_node); in FuseScaleOffsetToConvWeights()
171 new_nodes->push_back(bias_add_node); in FuseScaleOffsetToConvWeights()
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dgeneric_layout_optimizer_test.cc546 auto* bias_add_node = graph_view.GetNode("bias_add"); in TEST_F() local
547 ASSERT_NE(bias_add_node, nullptr); in TEST_F()
548 ASSERT_EQ(bias_add_node->NumRegularFanins(), 2); in TEST_F()
549 VerifyRegularFaninMatch(bias_add_node, 0, in TEST_F()
558 bias_add_node->GetName(), 0); in TEST_F()
Dremapper.cc2617 auto* bias_add_node = in AddFusedMatMulBiasAddAndGelu() local
2619 fused_node.add_input(bias_add_node->input(1)); in AddFusedMatMulBiasAddAndGelu()