|
|
|
@ -422,8 +422,9 @@ bool IsGraphOutputValueNodeOrParameter(const AnfNodePtr &output, const py::tuple
|
|
|
|
|
MS_EXCEPTION(UnknownError) << "When graph output is Parameter, it should be found in graph parameters";
|
|
|
|
|
}
|
|
|
|
|
size_t index = it - params.cbegin();
|
|
|
|
|
if (index >= args.size()) {
|
|
|
|
|
MS_EXCEPTION(UnknownError) << "Index " << index << " equal or larger than args size " << args.size() << ".";
|
|
|
|
|
if (index >= args.size() + func_graph->hyper_param_count()) {
|
|
|
|
|
MS_EXCEPTION(UnknownError) << "Index " << index << " equal or larger than args size " << args.size()
|
|
|
|
|
<< " add Parameter count " << func_graph->hyper_param_count() << ".";
|
|
|
|
|
}
|
|
|
|
|
*ret_val = args[index];
|
|
|
|
|
return true;
|
|
|
|
|