Home
last modified time | relevance | path

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

/external/XNNPACK/src/
Ddeconvolution-nhwc.c555 uint32_t log2_input_element_size, in setup_conv_path() argument
587 xnn_indirection_init_deconv2d(deconvolution_op, mr, log2_input_element_size); in setup_conv_path()
598 .kc = group_input_channels << log2_input_element_size, in setup_conv_path()
607 .ga_stride = group_input_channels << log2_input_element_size, in setup_conv_path()
610 …ide = input_height * input_width * deconvolution_op->input_pixel_stride << log2_input_element_size, in setup_conv_path()
660 uint32_t log2_input_element_size, in setup_subconv2d_path() argument
694 xnn_indirection_init_subconv2d(deconvolution_op, mr, log2_input_element_size); in setup_subconv2d_path()
722 .kc = group_input_channels << log2_input_element_size, in setup_subconv2d_path()
728 .ga_stride = group_input_channels << log2_input_element_size, in setup_subconv2d_path()
731 …ide = input_height * input_width * deconvolution_op->input_pixel_stride << log2_input_element_size, in setup_subconv2d_path()
[all …]
Dconvolution-nchw.c480 uint32_t log2_input_element_size, in setup_convolution2d_nchw() argument
584 .batched_a_stride = input_batch_stride << log2_input_element_size, in setup_convolution2d_nchw()
610 …ro_size = (input_width * convolution_op->group_input_channels << log2_input_element_size) + XNN_EX… in setup_convolution2d_nchw()
623 .input_batch_stride = input_batch_stride << log2_input_element_size, in setup_convolution2d_nchw()
627 .output_batch_stride = output_batch_stride << log2_input_element_size, in setup_convolution2d_nchw()
662 .input_channel_stride = input_height * input_width << log2_input_element_size, in setup_convolution2d_nchw()
663 .input_batch_stride = input_batch_stride << log2_input_element_size, in setup_convolution2d_nchw()
670 ….input_tuple_stride = convolution_op->ukernel.dwconv2d.input_width_tile << log2_input_element_size, in setup_convolution2d_nchw()
672 .input_pixel_stride = input_width << log2_input_element_size, in setup_convolution2d_nchw()
Dfully-connected-nc.c335 uint32_t log2_input_element_size, in setup_fully_connected_nc() argument
376 .k_scaled = input_channels << log2_input_element_size, in setup_fully_connected_nc()
377 …d_up_po2(input_channels, fully_connected_op->ukernel.gemm.kr) << log2_input_element_size) + bias_e… in setup_fully_connected_nc()
379 .a_stride = fully_connected_op->input_pixel_stride << log2_input_element_size, in setup_fully_connected_nc()
Dconvolution-nhwc.c726 uint32_t log2_input_element_size, in setup_convolution2d_nhwc() argument
811 .k_scaled = group_input_channels << log2_input_element_size, in setup_convolution2d_nhwc()
813 .a_stride = convolution_op->input_pixel_stride << log2_input_element_size, in setup_convolution2d_nhwc()
889 xnn_indirection_init_conv2d(convolution_op, mr, log2_input_element_size); in setup_convolution2d_nhwc()
898 .kc = group_input_channels << log2_input_element_size, in setup_convolution2d_nhwc()
907 .ga_stride = group_input_channels << log2_input_element_size, in setup_convolution2d_nhwc()
910 …tride = input_height * input_width * convolution_op->input_pixel_stride << log2_input_element_size, in setup_convolution2d_nhwc()
981 …_init_dwconv2d(convolution_op, valid_batch_size, step_height, step_width, log2_input_element_size); in setup_convolution2d_nhwc()
1015 .n = convolution_op->groups << log2_input_element_size, in setup_convolution2d_nhwc()
1017 .x_stride = convolution_op->input_pixel_stride << log2_input_element_size, in setup_convolution2d_nhwc()
Dmax-pooling-nhwc.c315 uint32_t log2_input_element_size, in setup_max_pooling2d() argument
379 … xnn_indirection_init_maxpool2d(max_pooling_op, step_height, step_width, log2_input_element_size); in setup_max_pooling2d()
398 …ide = (input_height * input_width * max_pooling_op->input_pixel_stride) << log2_input_element_size, in setup_max_pooling2d()