Searched defs:ms_func_graph (Results 1 – 3 of 3) sorted by relevance
226 auto ms_func_graph = cnode_morph->func_graph(); in ReplaceEquivdout() local
668 const ValuePtr &added_out, const FuncGraphPtr &ms_func_graph, in ReplaceNewTensorsInGradGraph()1626 void GradExecutor::MakeCNodeForMsFunction(const FuncGraphPtr &ms_func_graph, const py::args &args, in MakeCNodeForMsFunction()1687 void GradExecutor::MakeAdjointForMsFunction(const FuncGraphPtr &ms_func_graph, const FuncGraphPtr &… in MakeAdjointForMsFunction()3012 … const FuncGraphPtr &ms_func_graph, const FuncGraphPtr &grad_graph) { in GradMsFunctionInner()3068 FuncGraphPtr ms_func_graph = executor->GetFuncGraph(phase); in GradMsFunction() local
673 auto ms_func_graph = res->func_graph(); in EliminateForwardCNode() local