Searched refs:log2_input_element_size (Results 1 – 6 of 6) sorted by relevance
/external/XNNPACK/src/operators/ |
D | deconvolution-nhwc.c | 59 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 …]
|
D | convolution-nhwc.c | 82 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 …]
|
D | convolution-nchw.c | 521 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()
|
D | fully-connected-nc.c | 177 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()
|
D | average-pooling-nhwc.c | 409 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()
|
D | max-pooling-nhwc.c | 200 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()
|