Home
last modified time | relevance | path

Searched defs:packed_weights_size (Results 1 – 8 of 8) sorted by relevance

/external/XNNPACK/src/operators/
Dconvolution-nchw.c297 const size_t packed_weights_size = num_output_channel_blocks * sizeof(uint32_t) + in xnn_create_convolution2d_nchw_f32() local
411 const size_t packed_weights_size = groups * packed_group_output_channels * in xnn_create_convolution2d_nchw_f32() local
449 …const size_t packed_weights_size = groups * (kernel_height * kernel_width + 1 /* bias */) * sizeof… in xnn_create_convolution2d_nchw_f32() local
Dresize-bilinear-nchw.c152 const size_t packed_weights_size = sizeof(float) * (output_height * output_width * 2); in xnn_setup_resize_bilinear2d_nchw_f32() local
Dprelu-nc.c92 const size_t packed_weights_size = (channels << log2_weights_element_size) + XNN_EXTRA_BYTES; in create_prelu_nc() local
Dresize-bilinear-nhwc.c235 … const size_t packed_weights_size = (output_height * output_width * 2) << log2_weight_element_size; in setup_resize_bilinear2d_nhwc() local
Dfully-connected-nc.c121 …const size_t packed_weights_size = n_stride * (bias_element_size + (k_stride << log2_filter_elemen… in create_fully_connected_nc() local
Dconvolution-nhwc.c400 …const size_t packed_weights_size = ((UINT32_C(1) << log2_filter_element_size) + bias_element_size)… in create_convolution2d_nhwc() local
433 …const size_t packed_weights_size = ((kernel_size << log2_filter_element_size) + bias_element_size … in create_convolution2d_nhwc() local
/external/pytorch/aten/src/ATen/native/quantized/cpu/qnnpack/src/
Dconv-prepack.cc34 const size_t packed_weights_size = in PrePackConvWeights() local
Dconvolution.c230 const size_t packed_weights_size = in pytorch_qnnp_create_convolution_ndhwc_q8() local