Browse Source

fix ci bug

pull/14521/head
yeyunpeng2020 4 years ago
parent
commit
52a7de08d6
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