Home
last modified time | relevance | path

Searched refs:DepthwiseConvolution2DAttributes (Results 1 – 25 of 30) sorted by relevance

12

/external/tensorflow/tensorflow/lite/delegates/gpu/common/selectors/default/
Ddw_convolution_selector.cc27 const DepthwiseConvolution2DAttributes& attr, const GpuInfo& gpu_info, in SelectDWConvolutionAdreno()
39 const DepthwiseConvolution2DAttributes& attr, const GpuInfo& gpu_info, in SelectDWConvolutionPowerVR()
51 const DepthwiseConvolution2DAttributes& attr, const GpuInfo& gpu_info, in SelectDWConvolutionMali()
68 const DepthwiseConvolution2DAttributes& attr, const GpuInfo& gpu_info, in SelectDWConvolutionApple()
84 const DepthwiseConvolution2DAttributes& attr, const GpuInfo& gpu_info, in SelectDWConvolution()
/external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/special/
Ddepthwise_conv_plus_1x1_conv.h36 const DepthwiseConvolution2DAttributes& dw_attr,
41 const DepthwiseConvolution2DAttributes& dw_attr,
Ddepthwise_conv_plus_1x1_conv.cc27 void UploadWeights(const DepthwiseConvolution2DAttributes& dw_attr, in UploadWeights()
114 const DepthwiseConvolution2DAttributes& dw_attr, in GenerateCode()
235 const DepthwiseConvolution2DAttributes& dw_attr, in IsDepthwiseConvPlus1x1ConvSupported()
255 const DepthwiseConvolution2DAttributes& dw_attr, in CreateDepthwiseConvPlus1x1Conv()
/external/tensorflow/tensorflow/lite/delegates/gpu/common/transformations/
Dfuse_mul_to_conv.h48 DepthwiseConvolution2DAttributes* attr);
74 DepthwiseConvolution2DAttributes* attr);
Dfuse_add_to_conv.cc87 DepthwiseConvolution2DAttributes* conv_attr = in ApplyToNodesSequence()
88 absl::any_cast<DepthwiseConvolution2DAttributes>( in ApplyToNodesSequence()
123 DepthwiseConvolution2DAttributes* attr) { in FuseDepthwiseConvolution2DWithAdd()
Dfuse_mul_to_conv.cc78 DepthwiseConvolution2DAttributes* conv_attr = in ApplyToNodesSequence()
79 absl::any_cast<DepthwiseConvolution2DAttributes>( in ApplyToNodesSequence()
142 DepthwiseConvolution2DAttributes* conv_attr = in ApplyToNodesSequence()
143 absl::any_cast<DepthwiseConvolution2DAttributes>( in ApplyToNodesSequence()
198 DepthwiseConvolution2DAttributes* attr) { in FuseDepthwiseConvolution2DWithMultiply()
274 DepthwiseConvolution2DAttributes* attr) { in FuseMultiplyWithDepthwiseConvolution2D()
Dfuse_add_to_conv.h42 DepthwiseConvolution2DAttributes* attr);
Dadd_bias.cc79 auto& attr = absl::any_cast<DepthwiseConvolution2DAttributes&>( in ApplyToNode()
Dfuse_mul_to_conv_test.cc164 DepthwiseConvolution2DAttributes attr; in TEST()
248 DepthwiseConvolution2DAttributes attr; in TEST()
Dmerge_padding_with.cc109 MergePaddingWith2DOperation<DepthwiseConvolution2DAttributes>>( in NewMergePaddingWithDepthwiseConvolution()
Dfuse_add_to_conv_test.cc113 DepthwiseConvolution2DAttributes attr; in TEST()
/external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/
Ddepthwise_conv_test_util.cc33 DepthwiseConvolution2DAttributes attr; in DepthwiseConvSimpleWeightsTest()
70 DepthwiseConvolution2DAttributes attr; in DepthwiseConvNoMultiplierTest()
107 DepthwiseConvolution2DAttributes attr; in DepthwiseConvMultiplier2Test()
Ddepthwise_conv_3x3.h62 const DepthwiseConvolution2DAttributes& attr);
149 bool IsDepthwiseConv3x3Supported(const DepthwiseConvolution2DAttributes& attr);
153 const DepthwiseConvolution2DAttributes& attr);
Ddepthwise_conv_3x3_stride_h2.h56 const DepthwiseConvolution2DAttributes& attr, const GpuInfo& gpu_info);
145 const DepthwiseConvolution2DAttributes& attr, const GpuInfo& gpu_info);
148 const DepthwiseConvolution2DAttributes& attr);
Ddepthwise_conv_3x3_test_util.cc33 DepthwiseConvolution2DAttributes attr; in DepthwiseConv3x3SimpleWeightsTest()
71 DepthwiseConvolution2DAttributes attr; in DepthwiseConv3x3Test()
Ddepthwise_conv_3x3_stride_h2.cc210 const DepthwiseConvolution2DAttributes& attr, const GpuInfo& gpu_info) { in CreateDepthWiseConv3x3StrideH2()
235 const DepthwiseConvolution2DAttributes& attr) { in IsDepthWiseConv3x3StrideH2Supported()
Ddepthwise_conv_3x3_stride_h2_test_util.cc34 DepthwiseConvolution2DAttributes attr; in DepthWiseConv3x3StrideH2SimpleWeightsTest()
Ddepthwise_conv.h185 const DepthwiseConvolution2DAttributes& attr);
189 const DepthwiseConvolution2DAttributes& attr);
Ddepthwise_conv_3x3.cc302 bool IsDepthwiseConv3x3Supported(const DepthwiseConvolution2DAttributes& attr) { in IsDepthwiseConv3x3Supported()
313 const DepthwiseConvolution2DAttributes& attr) { in CreateDepthwiseConv3x3()
Ddepthwise_conv.cc238 const DepthwiseConvolution2DAttributes& attr) { in CreateDepthwiseConvolution2D()
274 const DepthwiseConvolution2DAttributes& attr) { in CreateDepthwiseConvolution2DDynamicWeights()
/external/tensorflow/tensorflow/lite/delegates/gpu/gl/kernels/
Ddepthwise_conv_test.cc39 DepthwiseConvolution2DAttributes attr; in TEST()
77 DepthwiseConvolution2DAttributes attr; in TEST()
116 DepthwiseConvolution2DAttributes attr; in TEST()
Ddepthwise_conv.cc46 absl::any_cast<const DepthwiseConvolution2DAttributes&>(ctx.op_attr); in GenerateCode()
/external/tensorflow/tensorflow/lite/delegates/gpu/common/selectors/
Ddw_convolution_selector.h29 const DepthwiseConvolution2DAttributes& attr, const GpuInfo& gpu_info,
Dsimple_selectors.h59 const DepthwiseConvolution2DAttributes& attr, const GpuInfo& gpu_info,
/external/tensorflow/tensorflow/lite/delegates/gpu/common/
Doperations.h325 struct DepthwiseConvolution2DAttributes : public Convolution2DAttributes {}; struct
331 const DepthwiseConvolution2DAttributes& attr);
341 const DepthwiseConvolution2DAttributes& attr);

12