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