From 8a6d5baeca1fe199ef34a7004d3372d5d38d8115 Mon Sep 17 00:00:00 2001 From: zhaozhenlong Date: Fri, 8 Jan 2021 15:41:29 +0800 Subject: [PATCH] find resize nn input tensor 2 num case --- .../lite/src/runtime/kernel/arm/fp32/resize_fp32.cc | 11 ----------- 1 file changed, 11 deletions(-) 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 61114a3720..36c9ca9503 100644 --- a/mindspore/lite/src/runtime/kernel/arm/fp32/resize_fp32.cc +++ b/mindspore/lite/src/runtime/kernel/arm/fp32/resize_fp32.cc @@ -198,17 +198,6 @@ int ResizeCPUKernel::RunImpl(int task_id) { break; } case static_cast(schema::ResizeMethod_NEAREST): { - if (in_tensors_.size() == lite::kDoubleNum && !const_shape_) { - auto out_shape = in_tensors_.at(1); - auto data = reinterpret_cast(out_shape->MutableData()); - if (data == nullptr) { - MS_LOG(ERROR) << "The out shape data is nullptr."; - return RET_NULL_PTR; - } else { - out_tensors_.at(0)->shape().at(1) = static_cast(data[0]); - out_tensors_.at(0)->shape().at(2) = static_cast(data[1]); - } - } ret = ResizeNearestNeighbor(input_data, output_data, input_shape.data(), out_tensors_[0]->shape().data(), calculate_, coordinate_transform_mode_, task_id, context_->thread_num_); break;