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.c484 uint32_t log2_output_element_size, in setup_conv_path() argument
534 .cm_stride = deconvolution_op->output_pixel_stride << log2_output_element_size, in setup_conv_path()
535 .cn_stride = nr << log2_output_element_size, in setup_conv_path()
538 .gc_stride = group_output_channels << log2_output_element_size, in setup_conv_path()
540 .bc_stride = output_size * deconvolution_op->output_pixel_stride << log2_output_element_size, in setup_conv_path()
541 .log2_csize = log2_output_element_size, in setup_conv_path()
607 uint32_t log2_output_element_size, in setup_subconv2d_path() argument
626 …nst size_t output_pixel_stride = deconvolution_op->output_pixel_stride << log2_output_element_size; in setup_subconv2d_path()
693 .cn_stride = nr << log2_output_element_size, in setup_subconv2d_path()
696 .gc_stride = group_output_channels << log2_output_element_size, in setup_subconv2d_path()
[all …]
Dconvolution-nhwc.c774 uint32_t log2_output_element_size, in setup_convolution2d_nhwc() argument
873 .cm_stride = convolution_op->output_pixel_stride << log2_output_element_size, in setup_convolution2d_nhwc()
874 .cn_stride = nr << log2_output_element_size, in setup_convolution2d_nhwc()
875 .cg_stride = group_output_channels << log2_output_element_size, in setup_convolution2d_nhwc()
876 .log2_csize = log2_output_element_size, in setup_convolution2d_nhwc()
982 .cm_stride = convolution_op->output_pixel_stride << log2_output_element_size, in setup_convolution2d_nhwc()
983 .cn_stride = nr << log2_output_element_size, in setup_convolution2d_nhwc()
986 .gc_stride = group_output_channels << log2_output_element_size, in setup_convolution2d_nhwc()
988 … .bc_stride = output_size * convolution_op->output_pixel_stride << log2_output_element_size, in setup_convolution2d_nhwc()
989 .log2_csize = log2_output_element_size, in setup_convolution2d_nhwc()
[all …]
Dfully-connected-nc.c166 uint32_t log2_output_element_size, in setup_fully_connected_nc() argument
212 .cm_stride = fully_connected_op->output_pixel_stride << log2_output_element_size, in setup_fully_connected_nc()
213 .cn_stride = nr << log2_output_element_size, in setup_fully_connected_nc()
214 .log2_csize = log2_output_element_size, in setup_fully_connected_nc()
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()
Dconvolution-nchw.c523 uint32_t log2_output_element_size, in setup_convolution2d_nchw() argument
566 …= (output_height * output_width * convolution_op->output_pixel_stride) << log2_output_element_size; in setup_convolution2d_nchw()
652 .output_height_stride = output_width << log2_output_element_size, in setup_convolution2d_nchw()
653 .output_channel_stride = output_height * output_width << log2_output_element_size, in setup_convolution2d_nchw()
703 .output_channel_stride = output_height * output_width << log2_output_element_size, in setup_convolution2d_nchw()
Dmax-pooling-nhwc.c199 uint32_t log2_output_element_size, in setup_max_pooling2d_nhwc() argument
294 …const size_t output_width_stride = max_pooling_op->output_pixel_stride << log2_output_element_size; in setup_max_pooling2d_nhwc()
310 .output_increment = output_width_stride - (channels << log2_output_element_size), in setup_max_pooling2d_nhwc()