Home
last modified time | relevance | path

Searched refs:OpParallelConfig (Results 1 – 6 of 6) sorted by relevance

/third_party/mindspore/mindspore/parallel/nn/
Dop_parallel_config.py42 class OpParallelConfig(_Config): class
123 default_dpmp_config = OpParallelConfig()
D__init__.py24 from .op_parallel_config import OpParallelConfig
Dloss.py26 from .op_parallel_config import default_dpmp_config, OpParallelConfig
68 if not isinstance(parallel_config, OpParallelConfig):
Dtransformer.py37 from .op_parallel_config import default_dpmp_config, _PipeLineConfig, OpParallelConfig, _Config, _c…
71 … self._dp_mp_config = OpParallelConfig(data_parallel=data_parallel, model_parallel=model_parallel)
318 parallel_config=_valid_type_checks([OpParallelConfig],
433 … parallel_config=_valid_type_checks([OpParallelConfig], "AttentionMask"))
681 parallel_config=_valid_type_checks([OpParallelConfig],
1147 parallel_config=_valid_type_checks([OpParallelConfig],
1444 parallel_config=_valid_type_checks([OpParallelConfig],
Dlayers.py34 from .op_parallel_config import default_dpmp_config, OpParallelConfig
488 … parallel_config=_valid_type_checks([OpParallelConfig], "FixedSparseAttention"))
/third_party/mindspore/tests/ut/python/parallel/
Dtest_parallel_transformer.py25 …VocabEmbedding, CrossEntropyLoss, OpParallelConfig, EmbeddingOpParallelConfig, FixedSparseAttention
463 sparse_attention_config = OpParallelConfig(model_parallel=8)
482 sparse_attention_config = OpParallelConfig(data_parallel=2, model_parallel=4)
501 sparse_attention_config = OpParallelConfig(data_parallel=4, model_parallel=2)
520 sparse_attention_config = OpParallelConfig(data_parallel=8, model_parallel=1)
635 parallel_test_config = OpParallelConfig(data_parallel=1, model_parallel=3)