Searched defs:grad_graph (Results 1 – 6 of 6) sorted by relevance
89 auto grad_graph = meta_graph->GenerateFuncGraph(args_abs); in GetPrimBpropGraph() local150 auto grad_graph = in GetFuncGraphBpropGraph() local
36 def grad_graph(grad_net, ms_inputs, out): member in ShapeFactory
167 static bool CallGradNodes(const FuncGraphPtr &graph, const FuncGraphPtr &grad_graph, in CallGradNodes()206 const auto &grad_graph = cnode->func_graph(); in FindGradGraph() local224 FuncGraphPtr grad_graph = FindGradGraph(root); in SetParameterStartForCellShare() local
487 auto grad_graph = ad::Grad(is_control_flow ? BasicClone(jit_forward_graph) : jit_forward_graph, in GetJitGradGraph() local
1298 FuncGraphPtr grad_graph; in IsolatedNodeAttach() local
805 void GraphExecutorPy::SetJitGradGraph(const FuncGraphPtr &grad_graph, const std::string &phase) { in SetJitGradGraph()