p12346870
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

!23849 [MS][LITE][CPU] nchw common conv bug fix 

Merge pull request !23849 from liuzhongkai/nchw_supp
This commit is contained in:
i-robot 2021-09-22 08:02:42 +00:00 committed by Gitee
parent 17981eb2cc 3d878451fe
commit 3330a630e8
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
mindspore/lite/src/runtime/kernel/arm/fp32/convolution_fp32.cc
View File

@ -119,8 +119,8 @@ int ConvolutionCPUKernel::RunImpl(int task_id) {
ConvFp32OutNC4HW4(ori_input_data, packed_input_, reinterpret_cast<float *>(packed_weight_),
reinterpret_cast<float *>(bias_data_), col_major_input_, tmp_output_, task_id, conv_param_);
#else
MS_LOG(ERROR) << "ConvFp32OutNC4HW4 not implemented.";
return RET_ERROR;
ConvFp32(ori_input_data, packed_input_, reinterpret_cast<float *>(packed_weight_),
reinterpret_cast<float *>(bias_data_), col_major_input_, tmp_output_, task_id, conv_param_);
#endif
}
return RET_OK;