hyf152
/
mindspore2022
forked from huawei/mindspore2022
0
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

!26554 [MSLITE] delegate conv fp16 weight bug 

Merge pull request !26554 from ling/r1.5
This commit is contained in:
i-robot 2021-11-19 09:26:18 +00:00 committed by Gitee
parent 0c4e7c3a31 8db46231c3
commit bde4563e04
3 changed files with 0 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
mindspore/lite/src/runtime/kernel/arm/fp16/convolution_1x1_fp16.cc
View File

@ -136,7 +136,6 @@ void Convolution1x1FP16CPUKernel::PackWeight() {
int Convolution1x1FP16CPUKernel::Init() {
CHECK_LESS_RETURN(in_tensors_.size(), 2);
CHECK_LESS_RETURN(out_tensors_.size(), 1);
UpdateOriginWeightAndBias();
#ifdef ENABLE_ARM64
if (out_tensors_.front()->format() == NC4HW4) {
row_tile_ = C16NUM;

1
mindspore/lite/src/runtime/kernel/arm/fp16/convolution_fp16.cc
View File

@ -95,7 +95,6 @@ int ConvolutionFP16CPUKernel::InitTmpBuffer() {
int ConvolutionFP16CPUKernel::Init() {
CHECK_LESS_RETURN(in_tensors_.size(), 2);
CHECK_LESS_RETURN(out_tensors_.size(), 1);
UpdateOriginWeightAndBias();
#ifdef ENABLE_ARM64
row_tile_ = C16NUM;
#else

1
mindspore/lite/src/runtime/kernel/arm/fp16/convolution_winograd_fp16.cc
View File

@ -143,7 +143,6 @@ int ConvolutionWinogradFP16CPUKernel::ConfigInputOutput() {
int ConvolutionWinogradFP16CPUKernel::Init() {
CHECK_LESS_RETURN(in_tensors_.size(), 2);
CHECK_LESS_RETURN(out_tensors_.size(), 1);
UpdateOriginWeightAndBias();
col_tile_ = C8NUM;
#ifdef ENABLE_ARM64
row_tile_ = C16NUM;