Home
last modified time | relevance | path

Searched refs:log2_output_element_size (Results 1 – 6 of 6) sorted by relevance

/external/XNNPACK/src/operators/
Ddeconvolution-nhwc.c557 uint32_t log2_output_element_size, in setup_conv_path() argument
607 .cm_stride = deconvolution_op->output_pixel_stride << log2_output_element_size, in setup_conv_path()
608 .cn_stride = nr << log2_output_element_size, in setup_conv_path()
611 .gc_stride = group_output_channels << log2_output_element_size, in setup_conv_path()
613 .bc_stride = output_size * deconvolution_op->output_pixel_stride << log2_output_element_size, in setup_conv_path()
614 .log2_csize = log2_output_element_size, in setup_conv_path()
680 uint32_t log2_output_element_size, in setup_subconv2d_path() argument
699 …nst size_t output_pixel_stride = deconvolution_op->output_pixel_stride << log2_output_element_size; in setup_subconv2d_path()
767 .cn_stride = nr << log2_output_element_size, in setup_subconv2d_path()
770 .gc_stride = group_output_channels << log2_output_element_size, in setup_subconv2d_path()
[all …]
Dconvolution-nhwc.c1046 uint32_t log2_output_element_size, in setup_convolution2d_nhwc() argument
1143 .cm_stride = convolution_op->output_pixel_stride << log2_output_element_size, in setup_convolution2d_nhwc()
1144 .cn_stride = nr << log2_output_element_size, in setup_convolution2d_nhwc()
1145 .cg_stride = group_output_channels << log2_output_element_size, in setup_convolution2d_nhwc()
1146 .log2_csize = log2_output_element_size, in setup_convolution2d_nhwc()
1253 .cm_stride = convolution_op->output_pixel_stride << log2_output_element_size, in setup_convolution2d_nhwc()
1254 .cn_stride = nr << log2_output_element_size, in setup_convolution2d_nhwc()
1257 .gc_stride = group_output_channels << log2_output_element_size, in setup_convolution2d_nhwc()
1259 … .bc_stride = output_size * convolution_op->output_pixel_stride << log2_output_element_size, in setup_convolution2d_nhwc()
1260 .log2_csize = log2_output_element_size, in setup_convolution2d_nhwc()
[all …]
Dfully-connected-nc.c180 uint32_t log2_output_element_size, in setup_fully_connected_nc() argument
226 .cm_stride = fully_connected_op->output_pixel_stride << log2_output_element_size, in setup_fully_connected_nc()
227 .cn_stride = nr << log2_output_element_size, in setup_fully_connected_nc()
228 .log2_csize = log2_output_element_size, in setup_fully_connected_nc()
Dconvolution-nchw.c524 uint32_t log2_output_element_size, in setup_convolution2d_nchw() argument
567 …= (output_height * output_width * convolution_op->output_pixel_stride) << log2_output_element_size; in setup_convolution2d_nchw()
653 .output_height_stride = output_width << log2_output_element_size, in setup_convolution2d_nchw()
654 .output_channel_stride = output_height * output_width << log2_output_element_size, in setup_convolution2d_nchw()
704 .output_channel_stride = output_height * output_width << log2_output_element_size, in setup_convolution2d_nchw()
Daverage-pooling-nhwc.c410 uint32_t log2_output_element_size, in setup_average_pooling2d() argument
493 .output_batch_stride = average_pooling_op->output_pixel_stride << log2_output_element_size, in setup_average_pooling2d()
542 …t size_t output_width_stride = average_pooling_op->output_pixel_stride << log2_output_element_size; in setup_average_pooling2d()
592 .output_increment = output_width_stride - (channels << log2_output_element_size), in setup_average_pooling2d()
619 .output_increment = output_width_stride - (channels << log2_output_element_size), in setup_average_pooling2d()
Dmax-pooling-nhwc.c201 uint32_t log2_output_element_size, in setup_max_pooling2d_nhwc() argument
296 …const size_t output_width_stride = max_pooling_op->output_pixel_stride << log2_output_element_size; in setup_max_pooling2d_nhwc()
312 .output_increment = output_width_stride - (channels << log2_output_element_size), in setup_max_pooling2d_nhwc()