Browse Source

fix resize bug

tags/v1.1.0
gongdaguo 5 years ago
parent
commit
d18a8cbb9c
1 changed files with 4 additions and 0 deletions
  1. +4
    -0
      mindspore/lite/src/runtime/kernel/arm/fp32/resize_fp32.cc

+ 4
- 0
mindspore/lite/src/runtime/kernel/arm/fp32/resize_fp32.cc View File

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


Loading…
Cancel
Save