|
|
|
@@ -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(); |
|
|
|
|