Searched defs:forward_graph (Results 1 – 13 of 13) sorted by relevance
249 void DoOverLapWay2(const FuncGraphManagerPtr &manager, const FuncGraphPtr &forward_graph, in DoOverLapWay2()264 void DoOverLap(const FuncGraphManagerPtr &manager, const FuncGraphPtr &forward_graph, in DoOverLap()316 auto forward_graph = each_graph; in OverlapGradMatmulAndGradAllreduce() local
204 auto forward_graph = each_graph; in OverlapRecomputeAllGatherAndFlashAttentionGrad() local
249 auto forward_graph = each_graph; in LabelFineGrainedInterleavedIndex() local
289 auto forward_graph = each_graph; in OverlapGradComm() local
415 auto forward_graph = each_graph; in CellReuseProcess() local
95 auto forward_graph = graph_executor->GetFuncGraph(phase); in GetFuncGraphBpropGraph() local
132 FuncGraphPtr BpropFuncGraphManager::GetFuncGraphBpropGraph(const FuncGraphPtr &forward_graph, in GetFuncGraphBpropGraph()
189 auto forward_graph = *(graph_set.begin()); in ProcessCommOpsFusion() local
148 for (const auto &forward_graph : forward_graph_set) { in InsertVirtualDataset() local
42 def forward_graph(self, ms_inputs): member in ShapeFactory
1226 FuncGraphPtr forward_graph = real_fn->func_graph(); in GenerateFuncGraph() local1546 FuncGraphPtr forward_graph = real_fn->func_graph(); in GenerateFuncGraph() local
2846 FuncGraphPtr GetCellReuseBackwardGraph(const FuncGraphPtr &forward_graph) { in GetCellReuseBackwardGraph()
303 FuncGraphPtr forward_graph = real_fn->func_graph(); in Run() local