Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/common/
H A 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
/aosp_15_r20/external/tensorflow/tensorflow/tools/graph_transforms/
H A 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
H A 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
H A Dfold_batch_norms.cc53 const NodeDef& conv_node = match.inputs[0].node; in FoldBatchNorms() local
H A Dflatten_atrous.cc55 const NodeDef& conv_node = match.inputs[0].node; in FlattenAtrousConv() local
/aosp_15_r20/external/tensorflow/tensorflow/core/grappler/optimizers/
H A 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
H A 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
H A Dconstant_folding.cc3522 NodeDef* conv_node = in MulConvPushDown() local
H A Dremapper.cc319 const NodeDef* conv_node) { in IsCpuCompatibleDataFormat()
/aosp_15_r20/external/ComputeLibrary/src/graph/mutators/
H A 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
H A 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
/aosp_15_r20/external/pytorch/torch/_inductor/fx_passes/
H A Dbinary_folding.py127 def _check_conv_and_broadcast_op(conv_node, other): argument
211 def _create_new_conv_node(graph, conv_node, binary_node, other): argument
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/common/transformations/
H A Dfuse_add_to_conv_test.cc61 auto conv_node = graph.NewNode(); in TEST() local
196 auto conv_node = graph.NewNode(); in TEST() local
H A Dmerge_padding_with_test.cc48 auto conv_node = graph.NewNode(); in TEST() local
93 auto conv_node = graph.NewNode(); in TEST() local
H A Dfuse_add_to_conv.cc56 auto& conv_node = *sequence[0]; in ApplyToNodesSequence() local
140 auto& conv_node = *sequence[1]; in ApplyToNodesSequence() local
H A Dfuse_mul_to_conv_test.cc61 auto conv_node = graph.NewNode(); in TEST() local
112 auto conv_node = graph.NewNode(); in TEST() local
H A Dfuse_mul_to_conv.cc43 auto& conv_node = *sequence[0]; in ApplyToNodesSequence() local
108 auto& conv_node = *sequence[1]; in ApplyToNodesSequence() local
/aosp_15_r20/external/pytorch/torch/csrc/jit/passes/
H A Dfold_conv_bn.cpp79 auto conv_node = in replaceConvBiasWithGetAttr() local
/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/
H A Dmkl_layout_pass.cc1128 const Node* conv_node; in GetPadOrConv2D() local
1177 const Node* conv_node; in GetPadOrFusedConv2D() local
/aosp_15_r20/external/tensorflow/tensorflow/core/transforms/constant_folding/
H A Dpass.cc2895 Operation *conv_node = in matchAndRewrite() local