Searched defs:forward_start (Results 1 – 4 of 4) sorted by relevance
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/graph_util/ |
| D | grad_accumulation_utils.cc | 211 std::unordered_map<int64_t, std::vector<CNodePtr>> *forward_start, in ExtractMicroBatchBorderNodes()
235 … const std::unordered_map<int64_t, std::vector<CNodePtr>> &forward_start, in ReorderGradAccumulation()
|
| D | pipeline_split_utils.cc | 695 void ReorderForForward(const std::vector<AnfNodePtr> &forward_start, const std::vector<AnfNodePtr> … in ReorderForForward()
1019 void GetBorderNode(std::vector<AnfNodePtr> *forward_start, std::vector<AnfNodePtr> *forward_end, in GetBorderNode()
1109 std::vector<AnfNodePtr> forward_start; in Reorder() local
1150 std::vector<AnfNodePtr> forward_start; in ReorderForPredict() local
|
| D | fold_pipeline_split_utils.cc | 185 static bool ComputeLastSegForwardEndIdx(const PipelinePair &forward_start, size_t curr_idx, int64_t… in ComputeLastSegForwardEndIdx()
477 std::vector<AnfNodePtr> forward_start; in FoldPipelineReorder() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/ |
| D | step_parallel.cc | 3166 std::unordered_map<int64_t, std::vector<CNodePtr>> forward_start; in ReorderForGradAccumulation() local
|