Searched refs:new_bias_size (Results 1 – 4 of 4) sorted by relevance
| /third_party/mindspore/mindspore/lite/micro/coder/opcoders/nnacl/int8/ |
| D | conv2d_3x3_int8_coder.cc | 65 int new_bias_size = oC4 * C4NUM * sizeof(int32_t); in InitWeightBias() local
66 …new_bias_addr_ = static_cast<int32_t *>(allocator_->Malloc(kNumberTypeInt32, new_bias_size, kOffli… in InitWeightBias()
68 …MS_CHECK_RET_CODE(memset_s(new_bias_addr_, new_bias_size, 0, new_bias_size), "memset_s new_bias_ad… in InitWeightBias()
71 …MS_CHECK_RET_CODE(memcpy_s(new_bias_addr_, new_bias_size, ori_bias_addr, output_channel * sizeof(i… in InitWeightBias()
|
| /third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp32/ |
| D | convolution_winograd_fp32.cc | 233 size_t new_bias_size = UP_ROUND(out_channel, C4NUM) * sizeof(float); in MallocWeightBiasData() local
235 bias_data_ = malloc(new_bias_size); in MallocWeightBiasData()
241 memset(bias_data_, 0, new_bias_size); in MallocWeightBiasData()
|
| /third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/int8/ |
| D | convolution_3x3_int8.cc | 114 size_t new_bias_size = oC4 * C4NUM * sizeof(int32_t); in InitWeightBias() local
115 bias_data_ = reinterpret_cast<int32_t *>(malloc(new_bias_size)); in InitWeightBias()
120 memset(bias_data_, 0, new_bias_size); in InitWeightBias()
|
| /third_party/mindspore/mindspore/lite/tools/optimizer/fusion/ |
| D | conv_conv_fusion.cc | 87 int new_bias_size = down_weight_shape[0]; in GenNewConvBias() local
88 …auto tensor_info = lite::CreateTensorInfo(nullptr, 0, {new_bias_size}, up_bias_param->data_type()); in GenNewConvBias()
95 if (memset_s(new_bias_data, tensor_info->Size(), 0, new_bias_size * sizeof(float)) != EOK) { in GenNewConvBias()
100 for (int i = 0; i < new_bias_size; i++) { in GenNewConvBias()
|