From d18a8cbb9cb4cc5310e15d6a4220ddfbcf9f4a5f Mon Sep 17 00:00:00 2001
From: gongdaguo <gongdaguo@huawei.com>
Date: Wed, 16 Dec 2020 16:04:02 +0800
Subject: [PATCH] fix resize bug

---
 mindspore/lite/src/runtime/kernel/arm/fp32/resize_fp32.cc | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/mindspore/lite/src/runtime/kernel/arm/fp32/resize_fp32.cc b/mindspore/lite/src/runtime/kernel/arm/fp32/resize_fp32.cc
index 43f2ca43c1..a3278de069 100644
--- a/mindspore/lite/src/runtime/kernel/arm/fp32/resize_fp32.cc
+++ b/mindspore/lite/src/runtime/kernel/arm/fp32/resize_fp32.cc
@@ -42,6 +42,10 @@ int ResizeCPUKernel::Init() {
 int ResizeCPUKernel::ReSize() {
   int ret = RET_OK;
   if (method_ == static_cast<int>(schema::ResizeMethod_LINEAR)) {
+    if (!const_shape_) {
+      new_height_ = out_tensors_.at(0)->shape()[1];
+      new_width_ = out_tensors_.at(0)->shape()[2];
+    }
     FreeTmpBuffer();
     ret = MallocTmpBuffer();
     if (ret != RET_OK) {