Fix error type of inputs of ge graph is not tensor

This commit is contained in:
fary86 2020-04-27 10:17:05 +08:00
parent e213f2a435
commit 0f64f63ce2
1 changed files with 1 additions and 1 deletions

View File

@ -465,7 +465,7 @@ void ProcessGeArg(const std::map<std::string, ExecutorInfoPtr> &info, const py::
if (converted->isa<tensor::Tensor>()) {
inputs->push_back(converted->cast<tensor::TensorPtr>());
} else {
MS_LOG(EXCEPTION) << "Args " << converted->ToString() << " is not tensor";
MS_EXCEPTION(TypeError) << "Args " << converted->ToString() << " is not tensor";
}
}
}