Home
last modified time | relevance | path

Searched defs:backward_graph (Results 1 – 5 of 5) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/pass/
Doverlap_grad_comm.cc78 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
Doverlap_recompute_allgather_and_flashattention_grad.cc43 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
Doverlap_gradmatmul_and_gradallreduce.cc250 const FuncGraphPtr &backward_graph) { in DoOverLapWay2()
265 const FuncGraphPtr &backward_graph) { in DoOverLap()
317 auto backward_graph = GetCellReuseBackwardGraph(forward_graph); in OverlapGradMatmulAndGradAllreduce() local
Dlabel_fine_grained_interleaved_index.cc250 auto backward_graph = GetCellReuseBackwardGraph(forward_graph); in LabelFineGrainedInterleavedIndex() local
Dmicro_interleaved_order_control.cc416 auto backward_graph = GetCellReuseBackwardGraph(forward_graph); in CellReuseProcess() local