Fix error type of inputs of ge graph is not tensor

pull/719/head
fary86 5 years ago
parent e213f2a435
commit 0f64f63ce2

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

Loading…
Cancel
Save