Searched refs:log2_output_element_size (Results 1 – 5 of 5) sorted by relevance
| /aosp_15_r20/external/XNNPACK/src/operators/ |
| H A D | deconvolution-nhwc.c | 674 uint32_t log2_output_element_size, in setup_conv_path() argument
732 .cm_stride = deconvolution_op->output_pixel_stride << log2_output_element_size, in setup_conv_path()
733 .cn_stride = nr << log2_output_element_size, in setup_conv_path()
736 .gc_stride = group_output_channels << log2_output_element_size, in setup_conv_path()
738 .bc_stride = output_size * deconvolution_op->output_pixel_stride << log2_output_element_size, in setup_conv_path()
739 .log2_csize = log2_output_element_size, in setup_conv_path()
852 uint32_t log2_output_element_size, in setup_subconv2d_path() argument
878 …nst size_t output_pixel_stride = deconvolution_op->output_pixel_stride << log2_output_element_size; in setup_subconv2d_path()
960 .cn_stride = nr << log2_output_element_size, in setup_subconv2d_path()
963 .gc_stride = group_output_channels << log2_output_element_size, in setup_subconv2d_path()
[all …]
|
| H A D | convolution-nhwc.c | 1339 uint32_t log2_output_element_size, in setup_convolution2d_nhwc() argument
1463 .cm_stride = convolution_op->output_pixel_stride << log2_output_element_size, in setup_convolution2d_nhwc()
1464 .cn_stride = nr << log2_output_element_size, in setup_convolution2d_nhwc()
1465 .cg_stride = group_output_channels << log2_output_element_size, in setup_convolution2d_nhwc()
1466 .log2_csize = log2_output_element_size, in setup_convolution2d_nhwc()
1597 .cm_stride = convolution_op->output_pixel_stride << log2_output_element_size, in setup_convolution2d_nhwc()
1598 .cn_stride = nr << log2_output_element_size, in setup_convolution2d_nhwc()
1601 .gc_stride = group_output_channels << log2_output_element_size, in setup_convolution2d_nhwc()
1603 … .bc_stride = output_size * convolution_op->output_pixel_stride << log2_output_element_size, in setup_convolution2d_nhwc()
1604 .log2_csize = log2_output_element_size, in setup_convolution2d_nhwc()
[all …]
|
| H A D | convolution-nchw.c | 533 uint32_t log2_output_element_size, in setup_convolution2d_nchw() argument
582 …= (output_height * output_width * convolution_op->output_pixel_stride) << log2_output_element_size; in setup_convolution2d_nchw()
672 .output_height_stride = output_width << log2_output_element_size, in setup_convolution2d_nchw()
673 .output_channel_stride = output_height * output_width << log2_output_element_size, in setup_convolution2d_nchw()
727 .output_channel_stride = output_height * output_width << log2_output_element_size, in setup_convolution2d_nchw()
|
| H A D | fully-connected-nc.c | 198 uint32_t log2_output_element_size, in setup_fully_connected_nc() argument
258 .cm_stride = fully_connected_op->output_pixel_stride << log2_output_element_size, in setup_fully_connected_nc()
259 .cn_stride = nr << log2_output_element_size, in setup_fully_connected_nc()
260 .log2_csize = log2_output_element_size, in setup_fully_connected_nc()
|
| H A D | max-pooling-nhwc.c | 206 uint32_t log2_output_element_size, in setup_max_pooling2d_nhwc() argument
307 …const size_t output_width_stride = max_pooling_op->output_pixel_stride << log2_output_element_size; in setup_max_pooling2d_nhwc()
323 .output_increment = output_width_stride - (channels << log2_output_element_size), in setup_max_pooling2d_nhwc()
|