Home
last modified time | relevance | path

Searched defs:conv_node (Results 1 – 15 of 15) sorted by relevance

/external/tensorflow/tensorflow/tools/graph_transforms/
Dfuse_convolutions.cc49 const NodeDef& conv_node = match.node; in FuseResizePadAndConv() local
98 const NodeDef& conv_node = match.node; in FuseResizeAndConv() local
158 const NodeDef& conv_node = match.node; in FusePadAndConv() local
Dfold_batch_norms.cc53 const NodeDef& conv_node = match.inputs[0].node; in FoldBatchNorms() local
Dflatten_atrous.cc55 const NodeDef& conv_node = match.inputs[0].node; in FlattenAtrousConv() local
Dfold_old_batch_norms.cc110 const NodeDef& conv_node = conv_node_match.node; in FuseScaleOffsetToConvWeights() local
204 const NodeDef& conv_node = conv_node_match.node; in FuseBatchNormWithBatchToSpace() local
/external/tensorflow/tensorflow/lite/delegates/gpu/common/transformations/
Dmerge_padding_with_test.cc48 auto conv_node = graph.NewNode(); in TEST() local
93 auto conv_node = graph.NewNode(); in TEST() local
Dfuse_mul_to_conv_test.cc61 auto conv_node = graph.NewNode(); in TEST() local
112 auto conv_node = graph.NewNode(); in TEST() local
Dfuse_add_to_conv.cc56 auto& conv_node = *sequence[0]; in ApplyToNodesSequence() local
Dfuse_add_to_conv_test.cc61 auto conv_node = graph.NewNode(); in TEST() local
Dfuse_mul_to_conv.cc43 auto& conv_node = *sequence[0]; in ApplyToNodesSequence() local
108 auto& conv_node = *sequence[1]; in ApplyToNodesSequence() local
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dgeneric_layout_optimizer_test.cc290 auto* conv_node = graph_view.GetNode("Conv2D"); in TEST_F() local
309 auto* conv_node = graph_view.GetNode("Conv2D"); in TEST_F() local
332 auto* conv_node = graph_view.GetNode("Conv2D"); in TEST_F() local
351 auto* conv_node = graph_view.GetNode("Conv2D"); in TEST_F() local
374 auto* conv_node = graph_view.GetNode("Conv2D"); in TEST_F() local
Dlayout_optimizer_test.cc480 auto conv_node = node_map.GetNode("Conv2D"); in TEST_F() local
494 auto conv_node = node_map.GetNode("Conv2D"); in TEST_F() local
509 auto conv_node = node_map.GetNode("Conv2D"); in TEST_F() local
524 auto conv_node = node_map.GetNode("Conv2D"); in TEST_F() local
538 auto conv_node = node_map.GetNode("Conv2D"); in TEST_F() local
552 auto conv_node = node_map.GetNode("FusedBatchNormGrad"); in TEST_F() local
566 auto conv_node = node_map.GetNode("FusedBatchNormGrad"); in TEST_F() local
1231 auto conv_node = node_map.GetNode("Conv2D"); in TEST_F() local
Darithmetic_optimizer_test.cc1719 const NodeDef* conv_node = node_map.GetNode("Conv2D"); in TEST_F() local
1773 const NodeDef* conv_node = node_map.GetNode("Conv2D"); in TEST_F() local
Dconstant_folding.cc3463 NodeDef* conv_node = in MulConvPushDown() local
/external/tensorflow/tensorflow/lite/delegates/gpu/common/selectors/
Dspecial_selector.cc53 auto* conv_node = consumers[0]; in TryDepthwiseConvPlus1x1Conv() local
/external/tensorflow/tensorflow/core/common_runtime/
Dmkl_layout_pass.cc1179 const Node* conv_node; in GetPadOrConv2D() local
1228 const Node* conv_node; in GetPadOrFusedConv2D() local