Browse Source

[MSLITE] Fix the fp32 FC's resize bug

tags/v1.1.0
zhanyuan 5 years ago
parent
commit
ae0e3dab65
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