Home
last modified time | relevance | path

Searched defs:conv_node (Results 1 – 18 of 18) 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/
Dgpu_model_test_util.cc35 auto conv_node = graph.NewNode(); in TestLinkingConvolutionAndCosOp() local
138 auto conv_node = graph.NewNode(); in TestLinkingConvolution2InputMul2InputMul() local
277 auto conv_node = graph.NewNode(); in TestLinkingConvolution2InputBroadcastMul2InputMul() local
416 auto conv_node = graph.NewNode(); in TestLinkingConvolution2InputMul2InputBroadcastMul() local
555 auto conv_node = graph.NewNode(); in TestLinkingConvolution2InputMul2InputMulCos() local
703 auto conv_node = graph.NewNode(); in TestLinkingConvolutionFirstTanh2InputDiff() local
821 auto conv_node = graph.NewNode(); in TestLinkingConvolutionSecondTanh2InputDiff() local
948 auto conv_node = graph.NewNode(); in TestLinkingConvolutionFirstTanhSecondCos2InputDiff() local
1096 auto conv_node = graph.NewNode(); in TestLinkingComplex0() local
/external/ComputeLibrary/src/graph/mutators/
DGroupedConvolutionMutator.cpp92 … auto *conv_node = arm_compute::utils::cast::polymorphic_downcast<ConvolutionLayerNode *>(node); in create_grouped_convolution() local
138 … auto *conv_node = arm_compute::utils::cast::polymorphic_downcast<ConvolutionLayerNode *>(node); in mutate() local
DNodeFusionMutator.cpp86 …auto *conv_node = arm_compute::utils::cast::polymorphic_downcast<ConvolutionLayerNode *>(output_ed… in fuse_convolution_with_batch_normalization() local
269 auto *conv_node = arm_compute::utils::cast::polymorphic_downcast<N *>(output_edge->consumer()); in fuse_pad_with_convolution() local
545 …auto *conv_node = arm_compute::utils::cast::polymorphic_downcast<ConvolutionLayerNode *>(output_ed… in fuse_convolution_with_post_ops() local
644 …auto *conv_node = arm_compute::utils::cast::polymorphic_downcast<FusedConvolutionBatchNormalizatio… in fuse_convolution_batch_normalization_with_post_ops() local
/external/tensorflow/tensorflow/lite/delegates/gpu/common/transformations/
Dfuse_add_to_conv_test.cc61 auto conv_node = graph.NewNode(); in TEST() local
196 auto conv_node = graph.NewNode(); in TEST() local
Dmerge_padding_with_test.cc48 auto conv_node = graph.NewNode(); in TEST() local
93 auto conv_node = graph.NewNode(); in TEST() local
Dfuse_add_to_conv.cc56 auto& conv_node = *sequence[0]; in ApplyToNodesSequence() local
140 auto& conv_node = *sequence[1]; in ApplyToNodesSequence() 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_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.cc272 auto* conv_node = graph_view.GetNode("Conv2D"); in TEST_F() local
291 auto* conv_node = graph_view.GetNode("Conv2D"); in TEST_F() local
314 auto* conv_node = graph_view.GetNode("Conv2D"); in TEST_F() local
333 auto* conv_node = graph_view.GetNode("Conv2D"); in TEST_F() local
356 auto* conv_node = graph_view.GetNode("Conv2D"); in TEST_F() local
Darithmetic_optimizer_test.cc2178 const NodeDef* conv_node = node_map.GetNode("Conv2D"); in TEST_F() local
2232 const NodeDef* conv_node = node_map.GetNode("Conv2D"); in TEST_F() local
Dconstant_folding.cc3522 NodeDef* conv_node = in MulConvPushDown() local
Dremapper.cc319 const NodeDef* conv_node) { in IsCpuCompatibleDataFormat()
/external/tensorflow/tensorflow/core/common_runtime/
Dmkl_layout_pass.cc1128 const Node* conv_node; in GetPadOrConv2D() local
1177 const Node* conv_node; in GetPadOrFusedConv2D() local
/external/tensorflow/tensorflow/core/transforms/constant_folding/
Dpass.cc2895 Operation *conv_node = in matchAndRewrite() local