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.c59 uint32_t log2_input_element_size, in create_deconvolution2d_nhwc() argument
225 const size_t zero_size = (k_stride << log2_input_element_size) + XNN_EXTRA_BYTES; in create_deconvolution2d_nhwc()
554 uint32_t log2_input_element_size, in setup_conv_path() argument
589 xnn_indirection_init_deconv2d(deconvolution_op, mr, log2_input_element_size); in setup_conv_path()
600 .kc = group_input_channels << log2_input_element_size, in setup_conv_path()
609 .ga_stride = group_input_channels << log2_input_element_size, in setup_conv_path()
612 …ide = input_height * input_width * deconvolution_op->input_pixel_stride << log2_input_element_size, in setup_conv_path()
677 uint32_t log2_input_element_size, in setup_subconv2d_path() argument
698 const size_t input_pixel_stride = deconvolution_op->input_pixel_stride << log2_input_element_size; in setup_subconv2d_path()
743 xnn_indirection_init_subconv2d(deconvolution_op, mr, log2_input_element_size); in setup_subconv2d_path()
[all …]
Dconvolution-nhwc.c82 uint32_t log2_input_element_size, in create_convolution2d_nhwc() argument
324 zero_size = XNN_EXTRA_BYTES + (c_stride << log2_input_element_size); in create_convolution2d_nhwc()
426 zero_size = XNN_EXTRA_BYTES + (k_stride << log2_input_element_size); in create_convolution2d_nhwc()
1043 uint32_t log2_input_element_size, in setup_convolution2d_nhwc() argument
1136 .k_scaled = group_input_channels << log2_input_element_size, in setup_convolution2d_nhwc()
1138 .a_stride = convolution_op->input_pixel_stride << log2_input_element_size, in setup_convolution2d_nhwc()
1236 xnn_indirection_init_conv2d(convolution_op, mr, log2_input_element_size); in setup_convolution2d_nhwc()
1246 .kc = group_input_channels << log2_input_element_size, in setup_convolution2d_nhwc()
1255 .ga_stride = group_input_channels << log2_input_element_size, in setup_convolution2d_nhwc()
1258 …tride = input_height * input_width * convolution_op->input_pixel_stride << log2_input_element_size, in setup_convolution2d_nhwc()
[all …]
Dconvolution-nchw.c521 uint32_t log2_input_element_size, in setup_convolution2d_nchw() argument
566 …ide = (input_height * input_width * convolution_op->input_pixel_stride) << log2_input_element_size; in setup_convolution2d_nchw()
631 …ro_size = (input_width * convolution_op->group_input_channels << log2_input_element_size) + XNN_EX… in setup_convolution2d_nchw()
680 const size_t zero_size = (input_width << log2_input_element_size) + 2 * XNN_EXTRA_BYTES; in setup_convolution2d_nchw()
694 .input_width = input_width << log2_input_element_size, in setup_convolution2d_nchw()
698 .input_channel_stride = input_height * input_width << log2_input_element_size, in setup_convolution2d_nchw()
Dfully-connected-nc.c177 uint32_t log2_input_element_size, in setup_fully_connected_nc() argument
220 .k_scaled = input_channels << log2_input_element_size, in setup_fully_connected_nc()
221 …d_up_po2(input_channels, fully_connected_op->ukernel.gemm.kr) << log2_input_element_size) + bias_e… in setup_fully_connected_nc()
223 .a_stride = fully_connected_op->input_pixel_stride << log2_input_element_size, in setup_fully_connected_nc()
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.c200 uint32_t log2_input_element_size, in setup_max_pooling2d_nhwc() argument
285 … xnn_indirection_init_maxpool2d(max_pooling_op, step_height, step_width, log2_input_element_size); in setup_max_pooling2d_nhwc()
304 …ide = (input_height * input_width * max_pooling_op->input_pixel_stride) << log2_input_element_size, in setup_max_pooling2d_nhwc()