Home
last modified time | relevance | path

Searched refs:depthwise_convolution_2d (Results 1 – 4 of 4) sorted by relevance

/external/XNNPACK/src/subgraph/
Ddepthwise-convolution-2d.c164 node->params.depthwise_convolution_2d.input_padding_top = input_padding_top; in xnn_define_depthwise_convolution_2d()
165 node->params.depthwise_convolution_2d.input_padding_right = input_padding_right; in xnn_define_depthwise_convolution_2d()
166 node->params.depthwise_convolution_2d.input_padding_bottom = input_padding_bottom; in xnn_define_depthwise_convolution_2d()
167 node->params.depthwise_convolution_2d.input_padding_left = input_padding_left; in xnn_define_depthwise_convolution_2d()
168 node->params.depthwise_convolution_2d.kernel_height = kernel_height; in xnn_define_depthwise_convolution_2d()
169 node->params.depthwise_convolution_2d.kernel_width = kernel_width; in xnn_define_depthwise_convolution_2d()
170 node->params.depthwise_convolution_2d.subsampling_height = subsampling_height; in xnn_define_depthwise_convolution_2d()
171 node->params.depthwise_convolution_2d.subsampling_width = subsampling_width; in xnn_define_depthwise_convolution_2d()
172 node->params.depthwise_convolution_2d.dilation_height = dilation_height; in xnn_define_depthwise_convolution_2d()
173 node->params.depthwise_convolution_2d.dilation_width = dilation_width; in xnn_define_depthwise_convolution_2d()
[all …]
/external/XNNPACK/src/
Dsubgraph.c173 …if ((node->params.depthwise_convolution_2d.dilation_height | node->params.depthwise_convolution_2d in xnn_check_nchw_compatibility()
179 if (node->params.depthwise_convolution_2d.depth_multiplier != 1) { in xnn_check_nchw_compatibility()
182 …if (node->params.depthwise_convolution_2d.subsampling_height != node->params.depthwise_convolution… in xnn_check_nchw_compatibility()
185 switch (node->params.depthwise_convolution_2d.subsampling_height) { in xnn_check_nchw_compatibility()
192 …if (node->params.depthwise_convolution_2d.kernel_height != node->params.depthwise_convolution_2d.k… in xnn_check_nchw_compatibility()
195 switch (node->params.depthwise_convolution_2d.kernel_height) { in xnn_check_nchw_compatibility()
197 return node->params.depthwise_convolution_2d.input_padding_top == 1 && in xnn_check_nchw_compatibility()
198 node->params.depthwise_convolution_2d.input_padding_right == 1 && in xnn_check_nchw_compatibility()
199 node->params.depthwise_convolution_2d.input_padding_bottom == 1 && in xnn_check_nchw_compatibility()
200 …node->params.depthwise_convolution_2d.input_padding_left == 1 ? XNN_LAYOUT_FLAG_COMPATIBLE_NCHW : … in xnn_check_nchw_compatibility()
[all …]
Druntime.c335 node->params.depthwise_convolution_2d.input_padding_top, in xnn_create_runtime_v2()
336 node->params.depthwise_convolution_2d.input_padding_right, in xnn_create_runtime_v2()
337 node->params.depthwise_convolution_2d.input_padding_bottom, in xnn_create_runtime_v2()
338 node->params.depthwise_convolution_2d.input_padding_left, in xnn_create_runtime_v2()
339 node->params.depthwise_convolution_2d.kernel_height, in xnn_create_runtime_v2()
340 node->params.depthwise_convolution_2d.kernel_width, in xnn_create_runtime_v2()
341 node->params.depthwise_convolution_2d.subsampling_height, in xnn_create_runtime_v2()
342 node->params.depthwise_convolution_2d.subsampling_width, in xnn_create_runtime_v2()
343 node->params.depthwise_convolution_2d.dilation_height, in xnn_create_runtime_v2()
344 node->params.depthwise_convolution_2d.dilation_width, in xnn_create_runtime_v2()
[all …]
/external/XNNPACK/src/xnnpack/
Dsubgraph.h168 } depthwise_convolution_2d; member