Searched refs:_ps_context (Results 1 – 14 of 14) sorted by relevance
| /third_party/mindspore/mindspore/parallel/ |
| D | _ps_context.py | 21 _ps_context = None variable
42 global _ps_context
43 if _ps_context is None:
44 _ps_context = PSContext.get_instance()
45 return _ps_context
|
| D | _auto_parallel_context.py | 21 from mindspore.parallel._ps_context import _is_role_pserver
|
| /third_party/mindspore/mindspore/common/ |
| D | parameter.py | 30 from ..parallel._ps_context import _is_role_worker, _is_role_pserver, _is_role_sched, _clone_hash_t…
31 from ..parallel._ps_context import _reinsert_hash_table_size
32 from ..parallel._ps_context import _insert_weight_init_info, _insert_accumu_init_info
|
| D | api.py | 31 from ..parallel._ps_context import _is_role_pserver
|
| /third_party/mindspore/tests/st/ps/cmp_sparse_embedding/ |
| D | test_cmp_sparse_embedding.py | 29 from mindspore.parallel._ps_context import _is_role_pserver, _is_role_worker
|
| /third_party/mindspore/tests/st/ps/multi_full_ps/ |
| D | test_multi_full_ps.py | 26 from mindspore.parallel._ps_context import _is_role_pserver
|
| /third_party/mindspore/mindspore/communication/ |
| D | management.py | 17 from mindspore.parallel._ps_context import _is_role_pserver, _is_role_sched
|
| D | _comm_helper.py | 17 from mindspore.parallel._ps_context import _is_role_pserver, _is_role_sched
|
| /third_party/mindspore/mindspore/nn/layer/ |
| D | embedding.py | 26 from mindspore.parallel._ps_context import _is_role_worker, _get_ps_context
27 from mindspore.parallel._ps_context import _insert_hash_table_size, _set_cache_enable, _set_rank_id
|
| D | thor_layer.py | 28 from mindspore.parallel._ps_context import _is_role_worker, _get_ps_context
|
| /third_party/mindspore/mindspore/ |
| D | context.py | 31 from mindspore.parallel._ps_context import _set_ps_context, _get_ps_context, _reset_ps_context
|
| /third_party/mindspore/mindspore/train/callback/ |
| D | _checkpoint.py | 28 from mindspore.parallel._ps_context import _is_role_pserver, _get_ps_mode_rank
|
| /third_party/mindspore/mindspore/train/ |
| D | model.py | 31 from ..parallel._ps_context import _is_role_pserver, _is_role_sched
|
| /third_party/mindspore/mindspore/dataset/engine/ |
| D | datasets.py | 52 from mindspore.parallel._ps_context import _is_role_pserver, _is_role_sched
|