Searched refs:log2_input_element_size (Results 1 – 5 of 5) sorted by relevance
/aosp_15_r20/external/XNNPACK/src/operators/ |
H A D | convolution-nhwc.c | 71 size_t log2_input_element_size, in get_generated_gemm() argument 91 group_input_channels << log2_input_element_size, in get_generated_gemm() 113 size_t log2_input_element_size, in generate_gemms_up_to_max_mr() argument 122 log2_input_element_size, convolution_op->code_cache); in generate_gemms_up_to_max_mr() 126 log2_input_element_size, convolution_op->code_cache); in generate_gemms_up_to_max_mr() 136 size_t log2_input_element_size, in get_generated_igemm() argument 157 group_input_channels << log2_input_element_size, in get_generated_igemm() 178 size_t log2_input_element_size, in generate_igemms_up_to_max_mr() argument 188 log2_input_element_size, kernel_size, 1, convolution_op->code_cache); in generate_igemms_up_to_max_mr() 192 log2_input_element_size, kernel_size, mr, convolution_op->code_cache); in generate_igemms_up_to_max_mr() [all …]
|
H A D | deconvolution-nhwc.c | 50 uint32_t log2_input_element_size, in create_deconvolution2d_nhwc() argument 231 const size_t zero_size = (k_stride << log2_input_element_size) + XNN_EXTRA_BYTES; in create_deconvolution2d_nhwc() 671 uint32_t log2_input_element_size, in setup_conv_path() argument 713 xnn_indirection_init_deconv2d(deconvolution_op, mr, log2_input_element_size); in setup_conv_path() 725 .kc = group_input_channels << log2_input_element_size, in setup_conv_path() 734 .ga_stride = group_input_channels << log2_input_element_size, in setup_conv_path() 737 …ide = input_height * input_width * deconvolution_op->input_pixel_stride << log2_input_element_size, in setup_conv_path() 849 uint32_t log2_input_element_size, in setup_subconv2d_path() argument 877 const size_t input_pixel_stride = deconvolution_op->input_pixel_stride << log2_input_element_size; in setup_subconv2d_path() 937 xnn_indirection_init_subconv2d(deconvolution_op, mr, log2_input_element_size); in setup_subconv2d_path() [all …]
|
H A D | convolution-nchw.c | 530 uint32_t log2_input_element_size, in setup_convolution2d_nchw() argument 581 …ide = (input_height * input_width * convolution_op->input_pixel_stride) << log2_input_element_size; in setup_convolution2d_nchw() 650 …ro_size = (input_width * convolution_op->group_input_channels << log2_input_element_size) + XNN_EX… in setup_convolution2d_nchw() 703 const size_t zero_size = (input_width << log2_input_element_size) + 2 * XNN_EXTRA_BYTES; in setup_convolution2d_nchw() 717 .input_width = input_width << log2_input_element_size, in setup_convolution2d_nchw() 721 .input_channel_stride = input_height * input_width << log2_input_element_size, in setup_convolution2d_nchw()
|
H A D | fully-connected-nc.c | 195 uint32_t log2_input_element_size, in setup_fully_connected_nc() argument 251 .k_scaled = input_channels << log2_input_element_size, in setup_fully_connected_nc() 253 …y_connected_op->ukernel.gemm.kr * fully_connected_op->ukernel.gemm.sr) << log2_input_element_size), in setup_fully_connected_nc() 255 .a_stride = fully_connected_op->input_pixel_stride << log2_input_element_size, in setup_fully_connected_nc()
|
H A D | max-pooling-nhwc.c | 205 uint32_t log2_input_element_size, in setup_max_pooling2d_nhwc() argument 296 … xnn_indirection_init_maxpool2d(max_pooling_op, step_height, step_width, log2_input_element_size); in setup_max_pooling2d_nhwc() 315 …ide = (input_height * input_width * max_pooling_op->input_pixel_stride) << log2_input_element_size, in setup_max_pooling2d_nhwc()
|