Browse Source

!14521 fix resize bug

From: @yeyunpeng2020
Reviewed-by: @ddwsky,@zhanghaibo5
Signed-off-by: @ddwsky
pull/14521/MERGE
mindspore-ci-bot Gitee 5 years ago
parent
commit
4a093fc025
1 changed files with 2 additions and 2 deletions
  1. +2
    -2
      mindspore/lite/src/runtime/kernel/opencl/kernel/resize.cc

+ 2
- 2
mindspore/lite/src/runtime/kernel/opencl/kernel/resize.cc View File

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


Loading…
Cancel
Save