Browse Source

!13189 make npu some non 4d input output error not 0

From: @zhaozhenlong
Reviewed-by: @zhang_xue_tong,@zhanghaibo5
Signed-off-by: @zhang_xue_tong
tags/v1.2.0-rc1
mindspore-ci-bot Gitee 4 years ago
parent
commit
9ca5cd3129
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