Browse Source

!8470 [MSLITE] Fix the fp32 FC's resize bug

From: @zhanyuan1
Reviewed-by: @zhang_xue_tong,@zhanghaibo5
Signed-off-by: @zhang_xue_tong
tags/v1.1.0
mindspore-ci-bot Gitee 5 years ago
parent
commit
0ac5134e7b
1 changed files with 2 additions and 0 deletions
  1. +2
    -0
      mindspore/lite/src/runtime/kernel/arm/fp32/fullconnection_fp32.cc

+ 2
- 0
mindspore/lite/src/runtime/kernel/arm/fp32/fullconnection_fp32.cc View File

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


Loading…
Cancel
Save