Browse Source

make npu non 4d op error instead of output 0

tags/v1.2.0-rc1
zhaozhenlong 5 years ago
parent
commit
ff0551fef7
1 changed files with 5 additions and 5 deletions
  1. +5
    -5
      mindspore/lite/src/runtime/agent/npu/npu_executor.cc

+ 5
- 5
mindspore/lite/src/runtime/agent/npu/npu_executor.cc View File

@@ -143,11 +143,11 @@ int NPUExecutor::Run(const std::vector<Tensor *> &in_tensors, const std::vector<
outputs_visited[index] = true;
break;
}
if (index == out_tensors.size()) {
MS_LOG(ERROR) << "Can't find corresponding ms lite tensor of " << i << " output tensor for npu executor "
<< model_name_;
return RET_ERROR;
}
}
if (index == out_tensors.size()) {
MS_LOG(ERROR) << "Can't find corresponding ms lite tensor of " << i << " output tensor for npu executor "
<< model_name_;
return RET_ERROR;
}
}
return RET_OK;


Loading…
Cancel
Save