Searched refs:PipelineTransformer (Results 1 – 3 of 3) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/frontend/parallel/pipeline_transformer/ |
| D | pipeline_transformer.cc | 58 void PipelineTransformer::MainGraph() { in MainGraph()
76 ValuePtr PipelineTransformer::SetMicroBatch(const AnfNodePtr &node, int64_t micro_size) { in SetMicroBatch()
91 bool PipelineTransformer::NeedGrad(const CNodePtr &cnode, const CNodePtr &graph_cnode) { in NeedGrad()
118 bool PipelineTransformer::LabelParameterStart(const FuncGraphPtr &graph, const CNodePtr &graph_cnod… in LabelParameterStart()
146 void PipelineTransformer::LabelMicroBatch() { in LabelMicroBatch()
178 void PipelineTransformer::CreateForwardGroup() { in CreateForwardGroup()
194 void PipelineTransformer::Coloring() { in Coloring()
234 void PipelineTransformer::BroadCastColoring() { in BroadCastColoring()
261 bool PipelineTransformer::IsPipelineCareNode(const CNodePtr &cnode) { in IsPipelineCareNode()
277 CNodePtr PipelineTransformer::GraphOutNode(const AnfNodePtr &node, int tuple_index) { in GraphOutNode()
[all …]
|
| D | pipeline_transformer.h | 41 class PipelineTransformer {
43 …PipelineTransformer(const FuncGraphManagerPtr &manager, int stage, const FuncGraphPtr &root, int64… in PipelineTransformer() function
52 virtual ~PipelineTransformer() = default;
|
| /third_party/mindspore/mindspore/ccsrc/pipeline/jit/ |
| D | pipeline_split.cc | 110 …std::make_shared<parallel::PipelineTransformer>(manager, stage, root, global_rank, per_stage_rank_… in PipelineSplit()
|