fix resize bug

4 years ago · d18a8cbb9c
parent c624b04b24
commit d18a8cbb9c
1 changed files with 4 additions and 0 deletions
--- 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) {