Home
last modified time | relevance | path

Searched refs:dilation_width (Results 1 – 25 of 35) sorted by relevance

12

/external/XNNPACK/src/subgraph/
Ddepthwise-convolution-2d.c27 uint32_t dilation_width, in xnn_define_depthwise_convolution_2d() argument
58 if (dilation_width == 0 || dilation_height == 0) { in xnn_define_depthwise_convolution_2d()
61 … xnn_node_type_to_string(xnn_node_type_depthwise_convolution_2d), dilation_width, dilation_height); in xnn_define_depthwise_convolution_2d()
123 const uint32_t padding_width = (kernel_width - 1) * dilation_width; in xnn_define_depthwise_convolution_2d()
173 node->params.depthwise_convolution_2d.dilation_width = dilation_width; in xnn_define_depthwise_convolution_2d()
Dmax-pooling-2d.c27 uint32_t dilation_width, in xnn_define_max_pooling_2d() argument
63 if (dilation_height == 0 || dilation_width == 0) { in xnn_define_max_pooling_2d()
66 xnn_node_type_to_string(xnn_node_type_max_pooling_2d), dilation_width, dilation_height); in xnn_define_max_pooling_2d()
132 node->params.pooling_2d.dilation_width = dilation_width; in xnn_define_max_pooling_2d()
Dconvolution-2d.c27 uint32_t dilation_width, in xnn_define_convolution_2d() argument
59 if (dilation_width == 0 || dilation_height == 0) { in xnn_define_convolution_2d()
62 xnn_node_type_to_string(xnn_node_type_convolution_2d), dilation_width, dilation_height); in xnn_define_convolution_2d()
131 const uint32_t padding_width = (kernel_width - 1) * dilation_width; in xnn_define_convolution_2d()
181 node->params.convolution_2d.dilation_width = dilation_width; in xnn_define_convolution_2d()
Ddeconvolution-2d.c29 uint32_t dilation_width, in xnn_define_deconvolution_2d() argument
61 if (dilation_width == 0 || dilation_height == 0) { in xnn_define_deconvolution_2d()
64 xnn_node_type_to_string(xnn_node_type_deconvolution_2d), dilation_width, dilation_height); in xnn_define_deconvolution_2d()
153 node->params.deconvolution_2d.dilation_width = dilation_width; in xnn_define_deconvolution_2d()
/external/XNNPACK/src/operators/
Dmax-pooling-nhwc.c55 uint32_t dilation_width, in create_max_pooling2d_nhwc() argument
110 if (dilation_height == 0 || dilation_width == 0) { in create_max_pooling2d_nhwc()
113 xnn_operator_type_to_string(operator_type), dilation_width, dilation_height); in create_max_pooling2d_nhwc()
172 max_pooling_op->dilation_width = dilation_width; in create_max_pooling2d_nhwc()
237 …t effective_kernel_width = (max_pooling_op->kernel_width - 1) * max_pooling_op->dilation_width + 1; in setup_max_pooling2d_nhwc()
255 max_pooling_op->dilation_width, in setup_max_pooling2d_nhwc()
267 …max_pooling_op->dilation_width > 1 ? pooling_width : min(max_pooling_op->stride_width, pooling_wid… in setup_max_pooling2d_nhwc()
334 uint32_t dilation_width, in xnn_create_max_pooling2d_nhwc_u8() argument
355 dilation_height, dilation_width, in xnn_create_max_pooling2d_nhwc_u8()
373 uint32_t dilation_width, in xnn_create_max_pooling2d_nhwc_f32() argument
[all …]
Dconvolution-nhwc.c71 uint32_t dilation_width, in create_convolution2d_nhwc() argument
139 if (dilation_width == 0 || dilation_height == 0) { in create_convolution2d_nhwc()
142 xnn_operator_type_to_string(operator_type), dilation_width, dilation_height); in create_convolution2d_nhwc()
394 convolution_op->dilation_width = dilation_width; in create_convolution2d_nhwc()
428 uint32_t dilation_width, in xnn_create_convolution2d_nhwc_qu8() argument
495 dilation_height, dilation_width, in xnn_create_convolution2d_nhwc_qu8()
526 uint32_t dilation_width, in xnn_create_convolution2d_nhwc_qs8() argument
589 dilation_height, dilation_width, in xnn_create_convolution2d_nhwc_qs8()
620 uint32_t dilation_width, in xnn_create_convolution2d_nhwc_f16() argument
670 dilation_height, dilation_width, in xnn_create_convolution2d_nhwc_f16()
[all …]
Ddeconvolution-nhwc.c51 uint32_t dilation_width, in create_deconvolution2d_nhwc() argument
100 if (dilation_width == 0 || dilation_height == 0) { in create_deconvolution2d_nhwc()
103 xnn_operator_type_to_string(operator_type), dilation_width, dilation_height); in create_deconvolution2d_nhwc()
178 …if (max(stride_height, stride_width) > 1 && max(dilation_height, dilation_width) == 1 && stride_wi… in create_deconvolution2d_nhwc()
255 deconvolution_op->dilation_width = dilation_width; in create_deconvolution2d_nhwc()
277 const uint32_t padding_width = (kernel_width - 1) * dilation_width; in create_deconvolution2d_nhwc()
311 uint32_t dilation_width, in xnn_create_deconvolution2d_nhwc_qu8() argument
378 dilation_height, dilation_width, in xnn_create_deconvolution2d_nhwc_qu8()
404 uint32_t dilation_width, in xnn_create_deconvolution2d_nhwc_f32() argument
456 dilation_height, dilation_width, in xnn_create_deconvolution2d_nhwc_f32()
[all …]
Dconvolution-nchw.c47 uint32_t dilation_width, in xnn_create_convolution2d_nchw_f32() argument
85 if (dilation_width == 0 || dilation_height == 0) { in xnn_create_convolution2d_nchw_f32()
88 …xnn_operator_type_to_string(xnn_operator_type_convolution_nchw_f32), dilation_width, dilation_heig… in xnn_create_convolution2d_nchw_f32()
175 …ol is_3x3 = kernel_width == 3 && kernel_height == 3 && dilation_height == 1 && dilation_width == 1; in xnn_create_convolution2d_nchw_f32()
176 …ol is_5x5 = kernel_width == 5 && kernel_height == 5 && dilation_height == 1 && dilation_width == 1; in xnn_create_convolution2d_nchw_f32()
215 …kernel_width, kernel_height, subsampling_width, subsampling_height, dilation_width, dilation_heigh… in xnn_create_convolution2d_nchw_f32()
487 convolution_op->dilation_width = dilation_width; in xnn_create_convolution2d_nchw_f32()
562 convolution_op->dilation_width, in setup_convolution2d_nchw()
/external/XNNPACK/src/
Dindirection.c36 const size_t dilation_width = op->dilation_width; in xnn_indirection_init_conv2d() local
56 … const size_t input_x = output_x * stride_width + kernel_x * dilation_width - input_padding_left; in xnn_indirection_init_conv2d()
96 const size_t dilation_width = op->dilation_width; in xnn_indirection_init_deconv2d() local
118 const size_t x = output_x + padding_left - kernel_x * dilation_width; in xnn_indirection_init_deconv2d()
217 const size_t dilation_width = op->dilation_width; in xnn_indirection_init_dwconv2d() local
227 … const size_t input_x = output_x * stride_width + kernel_x * dilation_width - input_padding_left; in xnn_indirection_init_dwconv2d()
267 const size_t dilation_width = op->dilation_width; in xnn_indirection_init_maxpool2d() local
271 const bool any_dilation = (dilation_height | dilation_width) > 1; in xnn_indirection_init_maxpool2d()
276 const size_t adjusted_padding_left = input_padding_left % dilation_width; in xnn_indirection_init_maxpool2d()
294 safe_input_x += dilation_width; in xnn_indirection_init_maxpool2d()
[all …]
Dim2col.c21 size_t dilation_width, in xnn_im2col_conv2d() argument
36 …const size_t input_x = output_x * subsampling_width + kernel_x * dilation_width - input_padding_le… in xnn_im2col_conv2d()
/external/XNNPACK/test/
Dmax-pooling-operator-tester.h263 inline MaxPoolingOperatorTester& dilation(uint32_t dilation_height, uint32_t dilation_width) { in dilation() argument
265 assert(dilation_width >= 1); in dilation()
267 this->dilation_width_ = dilation_width; in dilation()
281 inline MaxPoolingOperatorTester& dilation_width(uint32_t dilation_width) { in dilation_width() function
282 assert(dilation_width >= 1); in dilation_width()
283 this->dilation_width_ = dilation_width; in dilation_width()
287 inline uint32_t dilation_width() const { in dilation_width() function
296 return (pooling_width() - 1) * dilation_width() + 1; in dilated_pooling_width()
471 const size_t ix = ox * stride_width() + px * dilation_width() - padding_left(); in TestU8()
496 dilation_height(), dilation_width(), in TestU8()
[all …]
Dsubgraph-tester.h79 uint32_t dilation_height, uint32_t dilation_width, uint32_t groups, in add_conv() argument
87 subsampling_height, subsampling_width, dilation_height, dilation_width, in add_conv()
102 uint32_t dilation_height, uint32_t dilation_width, in add_depthwise_conv() argument
109 subsampling_height, subsampling_width, dilation_height, dilation_width, in add_depthwise_conv()
Dconvolution-operator-tester.h251 inline ConvolutionOperatorTester& dilation(uint32_t dilation_height, uint32_t dilation_width) { in dilation() argument
253 assert(dilation_width >= 1); in dilation()
255 this->dilation_width_ = dilation_width; in dilation()
269 inline ConvolutionOperatorTester& dilation_width(uint32_t dilation_width) { in dilation_width() argument
270 assert(dilation_width >= 1); in dilation_width()
271 this->dilation_width_ = dilation_width; in dilation_width()
275 inline uint32_t dilation_width() const { in dilation_width() function
349 return (kernel_width() - 1) * dilation_width() + 1; in dilated_kernel_width()
559 … const size_t ix = ox * subsampling_width() + kx * dilation_width() - padding_left(); in TestNHWCxQS8()
583 … const size_t ix = ox * subsampling_width() + kx * dilation_width() - padding_left(); in TestNHWCxQS8()
[all …]
Ddeconvolution-operator-tester.h283 inline DeconvolutionOperatorTester& dilation(uint32_t dilation_height, uint32_t dilation_width) { in dilation() argument
285 assert(dilation_width >= 1); in dilation()
287 this->dilation_width_ = dilation_width; in dilation()
301 inline DeconvolutionOperatorTester& dilation_width(uint32_t dilation_width) { in dilation_width() argument
302 assert(dilation_width >= 1); in dilation_width()
303 this->dilation_width_ = dilation_width; in dilation_width()
307 inline uint32_t dilation_width() const { in dilation_width() function
381 return (kernel_width() - 1) * dilation_width() + 1; in dilated_kernel_width()
534 const size_t x = ox + padding_left() - kx * dilation_width(); in TestQU8()
579 dilation_height(), dilation_width(), in TestQU8()
[all …]
Ddeconvolution-nhwc.cc770 for (size_t dilation_width = 2; dilation_width <= 3; dilation_width++) { variable
775 .dilation_width(dilation_width)
803 .dilation_width(3)
1101 for (size_t dilation_width = 2; dilation_width <= 3; dilation_width++) { in TEST() local
1106 .dilation_width(dilation_width) in TEST()
1136 .dilation_width(3) in TEST()
1440 for (size_t dilation_width = 2; dilation_width <= 3; dilation_width++) { in TEST() local
1446 .dilation_width(dilation_width) in TEST()
1476 .dilation_width(3) in TEST()
1793 for (size_t dilation_width = 2; dilation_width <= 3; dilation_width++) { in TEST() local
[all …]
/external/XNNPACK/src/xnnpack/
Dsubgraph.h133 uint32_t dilation_width; member
150 uint32_t dilation_width; member
165 uint32_t dilation_width; member
182 uint32_t dilation_width; member
Dim2col.h24 size_t dilation_width,
Doperator.h203 uint32_t dilation_width; member
/external/tensorflow/tensorflow/lite/kernels/
Dconv3d_test.cc37 int32_t dilation_depth = 1, int32_t dilation_width = 1, in Conv3dOpModel() argument
47 dilation_width, dilation_height) in Conv3dOpModel()
57 int32_t dilation_depth = 1, int32_t dilation_width = 1, in Conv3dOpModel() argument
66 dilation_width, dilation_height) in Conv3dOpModel()
Dconv3d.cc134 runtime_params.dilation_width = params->dilation_width_factor; in Eval()
/external/tensorflow/tensorflow/lite/delegates/xnnpack/
Dconv_2d_tester.h135 inline Conv2DTester& DilationWidth(int32_t dilation_width) { in DilationWidth() argument
136 EXPECT_GT(dilation_width, 0); in DilationWidth()
137 dilation_width_ = dilation_width; in DilationWidth()
Ddepthwise_conv_2d_tester.h139 inline DepthwiseConv2DTester& DilationWidth(int32_t dilation_width) { in DilationWidth() argument
140 EXPECT_GT(dilation_width, 0); in DilationWidth()
141 dilation_width_ = dilation_width; in DilationWidth()
/external/tensorflow/tensorflow/lite/delegates/coreml/builders/
Dconvolution_op_builder.cc52 int dilation_width = 0; in Build() local
62 dilation_width = conv_params->dilation_width_factor; in Build()
74 dilation_width = depthwise_conv_params->dilation_width_factor; in Build()
108 layer_->mutable_convolution()->add_dilationfactor(dilation_width); in Build()
/external/XNNPACK/include/
Dxnnpack.h248 uint32_t dilation_width,
303 uint32_t dilation_width,
357 uint32_t dilation_width,
535 uint32_t dilation_width,
1236 uint32_t dilation_width,
1268 uint32_t dilation_width,
1413 uint32_t dilation_width,
1657 uint32_t dilation_width,
1832 uint32_t dilation_width,
1939 uint32_t dilation_width,
[all …]
/external/tensorflow/tensorflow/lite/kernels/internal/reference/
Dconv3d.h71 in_x_origin + params.dilation_width * filter_x; in Conv3D()

12