Home
last modified time | relevance | path

Searched defs:pipeline_segment (Results 1 – 3 of 3) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/parallel/_transformer/
Dop_parallel_config.py150 def __init__(self, pipeline_stage=1, micro_batch_num=1, pipeline_segment=1): argument
168 def pipeline_segment(self): member in _PipeLineConfig
172 def pipeline_segment(self, value): member in _PipeLineConfig
Dtransformer.py229 …_(self, data_parallel=1, model_parallel=1, expert_parallel=1, pipeline_stage=1, pipeline_segment=1, argument
315 def pipeline_segment(self): member in TransformerOpParallelConfig
319 def pipeline_segment(self, value): member in TransformerOpParallelConfig
/third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/nn/
Dcell.py348 def pipeline_segment(self): member in Cell
352 def pipeline_segment(self, value): member in Cell