Home
last modified time | relevance | path

Searched defs:conv_attr (Results 1 – 11 of 11) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/common/transformations/
H A Dfuse_mul_to_conv.cc66 Convolution2DAttributes* conv_attr = in ApplyToNodesSequence() local
72 ConvolutionTransposedAttributes* conv_attr = in ApplyToNodesSequence() local
78 DepthwiseConvolution2DAttributes* conv_attr = in ApplyToNodesSequence() local
84 FullyConnectedAttributes* conv_attr = in ApplyToNodesSequence() local
130 Convolution2DAttributes* conv_attr = in ApplyToNodesSequence() local
136 ConvolutionTransposedAttributes* conv_attr = in ApplyToNodesSequence() local
142 DepthwiseConvolution2DAttributes* conv_attr = in ApplyToNodesSequence() local
148 FullyConnectedAttributes* conv_attr = in ApplyToNodesSequence() local
H A Dfuse_add_to_conv.cc75 Convolution2DAttributes* conv_attr = in ApplyToNodesSequence() local
81 ConvolutionTransposedAttributes* conv_attr = in ApplyToNodesSequence() local
87 DepthwiseConvolution2DAttributes* conv_attr = in ApplyToNodesSequence() local
93 FullyConnectedAttributes* conv_attr = in ApplyToNodesSequence() local
159 Convolution2DAttributes* conv_attr = in ApplyToNodesSequence() local
H A Dfuse_add_to_conv_test.cc45 Convolution2DAttributes conv_attr; in TEST() local
186 Convolution2DAttributes conv_attr; in TEST() local
H A Dmerge_padding_with_test.cc53 Convolution2DAttributes conv_attr; in TEST() local
98 Convolution2DAttributes conv_attr; in TEST() local
H A Dfuse_mul_to_conv_test.cc45 Convolution2DAttributes conv_attr; in TEST() local
102 Convolution2DAttributes conv_attr; in TEST() local
H A Dmake_fully_connected.cc59 const auto& conv_attr = absl::any_cast<const Convolution2DAttributes&>( in ApplyToNode() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/common/
H A Dgpu_model_test_util.cc39 Convolution2DAttributes conv_attr; in TestLinkingConvolutionAndCosOp() local
142 Convolution2DAttributes conv_attr; in TestLinkingConvolution2InputMul2InputMul() local
281 Convolution2DAttributes conv_attr; in TestLinkingConvolution2InputBroadcastMul2InputMul() local
420 Convolution2DAttributes conv_attr; in TestLinkingConvolution2InputMul2InputBroadcastMul() local
559 Convolution2DAttributes conv_attr; in TestLinkingConvolution2InputMul2InputMulCos() local
707 Convolution2DAttributes conv_attr; in TestLinkingConvolutionFirstTanh2InputDiff() local
825 Convolution2DAttributes conv_attr; in TestLinkingConvolutionSecondTanh2InputDiff() local
952 Convolution2DAttributes conv_attr; in TestLinkingConvolutionFirstTanhSecondCos2InputDiff() local
1100 Convolution2DAttributes conv_attr; in TestLinkingComplex0() local
H A Dmodel_builder.cc711 Convolution2DAttributes conv_attr; in ResolveGroupedConvolution() local
1277 Convolution2DAttributes conv_attr; in Parse() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/special/
H A Ddepthwise_conv_plus_1x1_conv.cc128 void ThinPointwiseFuser::AddConvData(const Convolution2DAttributes& conv_attr) { in AddConvData()
261 Convolution2DAttributes* conv_attr = in IsNodeSupported() local
323 Convolution2DAttributes* conv_attr = in ReserveNode() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/common/selectors/
H A Doperation_selector.cc177 Convolution2DAttributes* conv_attr = in AddDynamicConv() local
185 ConvolutionTransposedAttributes* conv_attr = in AddDynamicConv() local
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/mkl/
H A Dmkl_fused_batch_norm_op_test.cc363 ops::Conv2D::Attrs conv_attr; in VerifyFusedBatchNormGradWithConv2D() local