|
|
|
@@ -230,6 +230,7 @@ const void ConvTransformFusion::CalNewWeightTensor(float *weight_data, int kerne |
|
|
|
if (ret != EOK) { |
|
|
|
MS_LOG(ERROR) << "memcpy error: " << ret; |
|
|
|
lite::ReturnCode::GetSingleReturnCode()->UpdateReturnCode(lite::RET_MEMORY_FAILED); |
|
|
|
delete[] tmp_weight_data; |
|
|
|
return; |
|
|
|
} |
|
|
|
|
|
|
|
@@ -247,6 +248,7 @@ const void ConvTransformFusion::CalNewBiasTensor(float *bias_data, int kernel_nu |
|
|
|
if (EOK != memset_s(tmp_bias_data, kernel_num * sizeof(float), 0, kernel_num * sizeof(float))) { |
|
|
|
MS_LOG(ERROR) << "memset bias data failed"; |
|
|
|
lite::ReturnCode::GetSingleReturnCode()->UpdateReturnCode(lite::RET_MEMORY_FAILED); |
|
|
|
delete[] tmp_bias_data; |
|
|
|
return; |
|
|
|
} |
|
|
|
for (int i = 0; i < kernel_num; i++) { |
|
|
|
|