From ae0e3dab65df922997d1d7a04b225c87222983bc Mon Sep 17 00:00:00 2001 From: zhanyuan Date: Wed, 11 Nov 2020 17:32:03 +0800 Subject: [PATCH] [MSLITE] Fix the fp32 FC's resize bug --- .../lite/src/runtime/kernel/arm/fp32/fullconnection_fp32.cc | 2 ++ 1 file changed, 2 insertions(+) diff --git a/mindspore/lite/src/runtime/kernel/arm/fp32/fullconnection_fp32.cc b/mindspore/lite/src/runtime/kernel/arm/fp32/fullconnection_fp32.cc index 481fa25003..5a75e98e2b 100644 --- a/mindspore/lite/src/runtime/kernel/arm/fp32/fullconnection_fp32.cc +++ b/mindspore/lite/src/runtime/kernel/arm/fp32/fullconnection_fp32.cc @@ -59,6 +59,8 @@ int FullconnectionCPUKernel::ReSize() { #ifdef ENABLE_ARM64 if (fc_param_->row_ == 1) { is_vector_input_ = true; + } else { + is_vector_input_ = false; } #endif if (in_tensors_.size() == 3) {