Searched defs:parallel_config (Results 1 – 5 of 5) sorted by relevance
| /third_party/mindspore/mindspore/parallel/nn/ |
| D | moe.py | 95 parallel_config=default_dpmp_config): argument 260 parallel_config=None): argument 318 parallel_config=None): argument
|
| D | transformer.py | 326 parallel_config=default_dpmp_config): argument 434 def __init__(self, seq_length, parallel_config=default_dpmp_config): argument 519 … def __init__(self, vocab_size, embedding_size, parallel_config=default_embedding_parallel_config, argument 695 parallel_config=default_dpmp_config): argument 1165 parallel_config=default_dpmp_config): argument 1462 parallel_config=default_dpmp_config): argument 1702 def _set_parallel_configure_for_layer(network, layer_id, offset, parallel_config, layers): argument 1887 parallel_config=default_transformer_config): argument 2079 parallel_config=default_transformer_config): argument 2297 parallel_config=default_transformer_config): argument
|
| D | loss.py | 66 def __init__(self, parallel_config=default_dpmp_config): argument
|
| D | layers.py | 496 parallel_config=default_dpmp_config): argument
|
| /third_party/mindspore/tests/ut/python/parallel/ |
| D | test_parallel_transformer.py | 60 def __init__(self, en_layer, de_layer, parallel_config): argument
|