Browse Source

find resize nn input tensor 2 num case

tags/v1.2.0-rc1
zhaozhenlong 5 years ago
parent
commit
8a6d5baeca
1 changed files with 0 additions and 11 deletions
  1. +0
    -11
      mindspore/lite/src/runtime/kernel/arm/fp32/resize_fp32.cc

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

@@ -198,17 +198,6 @@ int ResizeCPUKernel::RunImpl(int task_id) {
break;
}
case static_cast<int>(schema::ResizeMethod_NEAREST): {
if (in_tensors_.size() == lite::kDoubleNum && !const_shape_) {
auto out_shape = in_tensors_.at(1);
auto data = reinterpret_cast<int32_t *>(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<int64_t>(data[0]);
out_tensors_.at(0)->shape().at(2) = static_cast<int64_t>(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;


Loading…
Cancel
Save