Searched defs:backward_graph (Results 1 – 5 of 5) sorted by relevance
78 void InsertDwMatmulDepend(const FuncGraphPtr &backward_graph, const std::vector<CNodePtr> &dw_matmu… in InsertDwMatmulDepend()95 void InsertDependForDxAndGradComm(const FuncGraphPtr &backward_graph, const std::vector<CNodePtr> &… in InsertDependForDxAndGradComm()240 void OverlapDxAndGradComm(const FuncGraphPtr &backward_graph) { in OverlapDxAndGradComm()286 FuncGraphPtr backward_graph = graph; in OverlapGradComm() local
43 void AddDependForRecomputedAllGatherAndGradientReduceScatter(const FuncGraphPtr &backward_graph) { in AddDependForRecomputedAllGatherAndGradientReduceScatter()108 void OverlapRecomputeAGAndFlashAttentionGrad(const FuncGraphPtr &backward_graph) { in OverlapRecomputeAGAndFlashAttentionGrad()201 FuncGraphPtr backward_graph = graph; in OverlapRecomputeAllGatherAndFlashAttentionGrad() local
250 const FuncGraphPtr &backward_graph) { in DoOverLapWay2()265 const FuncGraphPtr &backward_graph) { in DoOverLap()317 auto backward_graph = GetCellReuseBackwardGraph(forward_graph); in OverlapGradMatmulAndGradAllreduce() local
250 auto backward_graph = GetCellReuseBackwardGraph(forward_graph); in LabelFineGrainedInterleavedIndex() local
416 auto backward_graph = GetCellReuseBackwardGraph(forward_graph); in CellReuseProcess() local