|
|
|
@@ -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; |
|
|
|
|