Home
last modified time | relevance | path

Searched refs:log2_input_element_size (Results 1 – 5 of 5) sorted by relevance

/aosp_15_r20/external/XNNPACK/src/operators/
H A Dconvolution-nhwc.c71 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 Ddeconvolution-nhwc.c50 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 Dconvolution-nchw.c530 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 Dfully-connected-nc.c195 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 Dmax-pooling-nhwc.c205 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()