Searched defs:jit_forward_graph (Results 1 – 1 of 1) sorted by relevance
42 … bool modify_output, const FuncGraphPtr &jit_forward_graph, ValuePtr *added_out_v) { in GetOpRunInfo()131 CNodePtr GetAddedNode(const FuncGraphPtr &jit_forward_graph) { in GetAddedNode()254 const FuncGraphPtr &jit_forward_graph, AnfNodePtrList *input_nodes) const { in GetWeightsNode() argument288 const FuncGraphPtr &jit_forward_graph, CNodePtr *jit_cnode) const { in MakeCNodeForJit() argument305 … const FuncGraphPtr &jit_forward_graph, const FuncGraphPtr &jit_grad_graph, in MakeAdjointForJit()473 auto jit_forward_graph = resource->func_graph(); in GetJitGradGraph() local509 FuncGraphPtr Jit::GetJitForwardGraphCNodeInfo(const FuncGraphPtr &jit_forward_graph) { in GetJitForwardGraphCNodeInfo()571 FuncGraphPtr jit_forward_graph = executor->GetFuncGraph(graph_phase_); in GradJit() local