Home
last modified time | relevance | path

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

/external/XNNPACK/src/operators/
Ddeconvolution-nhwc.c60 uint32_t log2_input_element_size, in create_deconvolution2d_nhwc() argument
235 const size_t zero_size = (k_stride << log2_input_element_size) + XNN_EXTRA_BYTES; in create_deconvolution2d_nhwc()
481 uint32_t log2_input_element_size, in setup_conv_path() argument
516 xnn_indirection_init_deconv2d(deconvolution_op, mr, log2_input_element_size); in setup_conv_path()
527 .kc = group_input_channels << log2_input_element_size, in setup_conv_path()
536 .ga_stride = group_input_channels << log2_input_element_size, in setup_conv_path()
539 …ide = input_height * input_width * deconvolution_op->input_pixel_stride << log2_input_element_size, in setup_conv_path()
604 uint32_t log2_input_element_size, in setup_subconv2d_path() argument
625 const size_t input_pixel_stride = deconvolution_op->input_pixel_stride << log2_input_element_size; in setup_subconv2d_path()
670 xnn_indirection_init_subconv2d(deconvolution_op, mr, log2_input_element_size); in setup_subconv2d_path()
[all …]
Dconvolution-nhwc.c80 uint32_t log2_input_element_size, in create_convolution2d_nhwc() argument
299 zero_size = XNN_EXTRA_BYTES + (c_stride << log2_input_element_size); in create_convolution2d_nhwc()
365 zero_size = XNN_EXTRA_BYTES + (k_stride << log2_input_element_size); in create_convolution2d_nhwc()
771 uint32_t log2_input_element_size, in setup_convolution2d_nhwc() argument
866 .k_scaled = group_input_channels << log2_input_element_size, in setup_convolution2d_nhwc()
868 .a_stride = convolution_op->input_pixel_stride << log2_input_element_size, in setup_convolution2d_nhwc()
966 xnn_indirection_init_conv2d(convolution_op, mr, log2_input_element_size); in setup_convolution2d_nhwc()
975 .kc = group_input_channels << log2_input_element_size, in setup_convolution2d_nhwc()
984 .ga_stride = group_input_channels << log2_input_element_size, in setup_convolution2d_nhwc()
987 …tride = input_height * input_width * convolution_op->input_pixel_stride << log2_input_element_size, in setup_convolution2d_nhwc()
[all …]
Dfully-connected-nc.c163 uint32_t log2_input_element_size, in setup_fully_connected_nc() argument
206 .k_scaled = input_channels << log2_input_element_size, in setup_fully_connected_nc()
207 …d_up_po2(input_channels, fully_connected_op->ukernel.gemm.kr) << log2_input_element_size) + bias_e… in setup_fully_connected_nc()
209 .a_stride = fully_connected_op->input_pixel_stride << log2_input_element_size, in setup_fully_connected_nc()
Dconvolution-nchw.c520 uint32_t log2_input_element_size, in setup_convolution2d_nchw() argument
565 …ide = (input_height * input_width * convolution_op->input_pixel_stride) << log2_input_element_size; in setup_convolution2d_nchw()
630 …ro_size = (input_width * convolution_op->group_input_channels << log2_input_element_size) + XNN_EX… in setup_convolution2d_nchw()
679 const size_t zero_size = (input_width << log2_input_element_size) + 2 * XNN_EXTRA_BYTES; in setup_convolution2d_nchw()
693 .input_width = input_width << log2_input_element_size, in setup_convolution2d_nchw()
697 .input_channel_stride = input_height * input_width << log2_input_element_size, in setup_convolution2d_nchw()
Daverage-pooling-nhwc.c409 uint32_t log2_input_element_size, in setup_average_pooling2d() argument
483 …t size_t input_stride_in_bytes = average_pooling_op->input_pixel_stride << log2_input_element_size; in setup_average_pooling2d()
532 …nn_indirection_init_dwconv2d(average_pooling_op, step_height, step_width, log2_input_element_size); in setup_average_pooling2d()
580 …e = input_height * input_width * average_pooling_op->input_pixel_stride << log2_input_element_size, in setup_average_pooling2d()
610 …e = input_height * input_width * average_pooling_op->input_pixel_stride << log2_input_element_size, in setup_average_pooling2d()
Dmax-pooling-nhwc.c198 uint32_t log2_input_element_size, in setup_max_pooling2d_nhwc() argument
283 … xnn_indirection_init_maxpool2d(max_pooling_op, step_height, step_width, log2_input_element_size); in setup_max_pooling2d_nhwc()
302 …ide = (input_height * input_width * max_pooling_op->input_pixel_stride) << log2_input_element_size, in setup_max_pooling2d_nhwc()