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) {