Searched defs:model_parallel (Results 1 – 2 of 2) sorted by relevance
| /third_party/mindspore/mindspore/parallel/nn/ |
| D | op_parallel_config.py | 57 def __init__(self, data_parallel=1, model_parallel=1): argument
73 def model_parallel(self): member in OpParallelConfig
77 def model_parallel(self, value): member in OpParallelConfig
|
| D | transformer.py | 70 def __init__(self, data_parallel=1, model_parallel=1, vocab_emb_dp=True): argument
84 def model_parallel(self): member in EmbeddingOpParallelConfig
88 def model_parallel(self, value): member in EmbeddingOpParallelConfig
144 …def __init__(self, data_parallel=1, model_parallel=1, pipeline_stage=1, micro_batch_num=1, recompu… argument
188 def model_parallel(self): member in TransformerOpParallelConfig
192 def model_parallel(self, value): member in TransformerOpParallelConfig
|