diff --git a/mindspore/lite/src/runtime/kernel/opencl/kernel/resize.cc b/mindspore/lite/src/runtime/kernel/opencl/kernel/resize.cc index 4e0e89b840..ecb6bc3bef 100644 --- a/mindspore/lite/src/runtime/kernel/opencl/kernel/resize.cc +++ b/mindspore/lite/src/runtime/kernel/opencl/kernel/resize.cc @@ -33,7 +33,7 @@ using mindspore::schema::PrimitiveType_Resize; namespace mindspore::kernel { int ResizeOpenCLKernel::CheckSpecs() { - if (in_tensors_.size() != 2 || out_tensors_.size() != 1) { + if (!(in_tensors_.size() == 1 || in_tensors_.size() == 2) || out_tensors_.size() != 1) { MS_LOG(ERROR) << "in size: " << in_tensors_.size() << ", out size: " << out_tensors_.size(); return RET_ERROR; } @@ -120,7 +120,7 @@ int ResizeOpenCLKernel::Run() { } int ResizeOpenCLKernel::PreProcess() { - if (Type() == PrimitiveType_Resize && !op_parameter_->infer_flag_) { + if (Type() == PrimitiveType_Resize && !op_parameter_->infer_flag_ && in_tensors_.size() == 2) { auto shape_tensor = in_tensors_[1]; if (!shape_tensor->IsConst()) { ocl_runtime_->SyncCommandQueue();