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()
|