Searched refs:bias_element_size (Results 1 – 6 of 6) sorted by relevance
| /external/ComputeLibrary/src/core/NEON/kernels/arm_conv/depthwise/interleaves/ |
| D | generic.cpp | 35 bool include_bias, size_t bias_element_size, in PackingArguments() argument
39 include_bias(include_bias), bias_element_size(bias_element_size), in PackingArguments()
62 const auto pack_size = (packing_args.include_bias ? packing_args.bias_element_size : 0) + in get_storage_size_generic()
104 biases += (biases == nullptr) ? 0 : packing_args.bias_element_size * args.channel_multiplier; in pack_parameters_generic()
126 memcpy(buffer, biases, todo * packing_args.bias_element_size); in pack_parameters_generic()
127 biases += todo * packing_args.bias_element_size; in pack_parameters_generic()
131 memset(buffer, 0, vl * packing_args.bias_element_size); in pack_parameters_generic()
134 buffer += vl * packing_args.bias_element_size; in pack_parameters_generic()
|
| D | generic.hpp | 42 const size_t bias_element_size; member
55 size_t bias_element_size,
|
| /external/XNNPACK/src/operators/ |
| D | deconvolution-nhwc.c | 52 uint32_t bias_element_size, in create_deconvolution2d_nhwc() argument
162 …hts_size = (((kernel_size * k_stride) << log2_filter_element_size) + bias_element_size) * n_stride; in create_deconvolution2d_nhwc()
167 (((kernel_size * k_stride) << log2_filter_element_size) + bias_element_size * subkernels); in create_deconvolution2d_nhwc()
186 …subconvolution_params->w_stride = bias_element_size + ((k_stride * subkernel_size) << log2_filter_… in create_deconvolution2d_nhwc()
673 uint32_t bias_element_size, in setup_conv_path() argument
720 const size_t w_stride = bias_element_size + in setup_conv_path()
851 uint32_t bias_element_size, in setup_subconv2d_path() argument
949 const size_t w_stride = stride_height * stride_width * bias_element_size + in setup_subconv2d_path()
1045 uint32_t bias_element_size, in setup_deconvolution2d_nhwc() argument
1113 … log2_input_element_size, log2_filter_element_size, bias_element_size, log2_output_element_size, in setup_deconvolution2d_nhwc()
[all …]
|
| D | fully-connected-nc.c | 36 uint32_t bias_element_size, in create_fully_connected_nc() argument
121 …const size_t packed_weights_size = n_stride * (bias_element_size + (k_stride << log2_filter_elemen… in create_fully_connected_nc()
197 uint32_t bias_element_size, in setup_fully_connected_nc() argument
252 .w_stride = bias_element_size + in setup_fully_connected_nc()
|
| D | convolution-nhwc.c | 218 uint32_t bias_element_size, in create_convolution2d_nhwc() argument
400 …t packed_weights_size = ((UINT32_C(1) << log2_filter_element_size) + bias_element_size) * c_stride; in create_convolution2d_nhwc()
433 … packed_weights_size = ((kernel_size << log2_filter_element_size) + bias_element_size + extra_weig… in create_convolution2d_nhwc()
467 …kernel->channel_tile * ((kernel_size << log2_filter_element_size) + bias_element_size + extra_weig… in create_convolution2d_nhwc()
469 …+ dwconv_ukernel->channel_tile * ((kernel_size << log2_filter_element_size) + bias_element_size))); in create_convolution2d_nhwc()
499 …ghts_size = ((kernel_size * k_stride << log2_filter_element_size) + bias_element_size + extra_weig… in create_convolution2d_nhwc()
583 … gemm_parameters->nr * ((kernel_size * k_stride << log2_filter_element_size) + bias_element_size)); in create_convolution2d_nhwc()
584 …hts_stride = (kernel_size * k_stride << log2_filter_element_size) + bias_element_size + extra_weig… in create_convolution2d_nhwc()
|
| D | convolution-nchw.c | 532 uint32_t bias_element_size, in setup_convolution2d_nchw() argument
724 .weights_channel_stride = bias_element_size + in setup_convolution2d_nchw()
|