Home
last modified time | relevance | path

Searched refs:bias_element_size (Results 1 – 4 of 4) sorted by relevance

/external/XNNPACK/src/operators/
Dfully-connected-nc.c35 uint32_t bias_element_size, in create_fully_connected_nc() argument
105 …const size_t packed_weights_size = n_stride * (bias_element_size + (k_stride << log2_filter_elemen… in create_fully_connected_nc()
165 uint32_t bias_element_size, in setup_fully_connected_nc() argument
207 …put_channels, fully_connected_op->ukernel.gemm.kr) << log2_input_element_size) + bias_element_size, in setup_fully_connected_nc()
Ddeconvolution-nhwc.c62 uint32_t bias_element_size, in create_deconvolution2d_nhwc() argument
483 uint32_t bias_element_size, in setup_conv_path() argument
522 const size_t w_stride = bias_element_size + in setup_conv_path()
606 uint32_t bias_element_size, in setup_subconv2d_path() argument
682 const size_t w_stride = stride_height * stride_width * bias_element_size + in setup_subconv2d_path()
777 uint32_t bias_element_size, in setup_deconvolution2d_nhwc() argument
853 … log2_input_element_size, log2_filter_element_size, bias_element_size, log2_output_element_size, in setup_deconvolution2d_nhwc()
868 … log2_input_element_size, log2_filter_element_size, bias_element_size, log2_output_element_size, in setup_deconvolution2d_nhwc()
Dconvolution-nhwc.c82 uint32_t bias_element_size, in create_convolution2d_nhwc() argument
242 …t packed_weights_size = ((UINT32_C(1) << log2_filter_element_size) + bias_element_size) * c_stride; in create_convolution2d_nhwc()
267 …t packed_weights_size = ((kernel_size << log2_filter_element_size) + bias_element_size) * c_stride; in create_convolution2d_nhwc()
311 …ights_size = ((kernel_size * k_stride << log2_filter_element_size) + bias_element_size) * n_stride; in create_convolution2d_nhwc()
773 uint32_t bias_element_size, in setup_convolution2d_nhwc() argument
853 const size_t w_stride = bias_element_size + in setup_convolution2d_nhwc()
970 …, convolution_op->ukernel.igemm.kr) * kernel_size << log2_filter_element_size) + bias_element_size; in setup_convolution2d_nhwc()
Dconvolution-nchw.c522 uint32_t bias_element_size, in setup_convolution2d_nchw() argument
700 .weights_channel_stride = bias_element_size + in setup_convolution2d_nchw()