Home
last modified time | relevance | path

Searched refs:_is_role_worker (Results 1 – 5 of 5) sorted by relevance

/third_party/mindspore/mindspore/common/
Dparameter.py30 from ..parallel._ps_context import _is_role_worker, _is_role_pserver, _is_role_sched, _clone_hash_t…
189 … if _is_in_parallel_mode() or _is_role_worker() or _is_role_sched() or _is_role_pserver():
224 if not(_is_role_worker() or _is_role_pserver() or _is_role_sched()):
290 if _is_role_worker() and self.cache_enable:
576 if _is_role_worker() and self.cache_enable:
593 self.init_mode.init is not None and (_is_role_worker() or _is_role_sched()):
667 if _is_role_worker():
/third_party/mindspore/tests/st/ps/cmp_sparse_embedding/
Dtest_cmp_sparse_embedding.py29 from mindspore.parallel._ps_context import _is_role_pserver, _is_role_worker
89 if _is_role_worker():
/third_party/mindspore/mindspore/nn/layer/
Dembedding.py26 from mindspore.parallel._ps_context import _is_role_worker, _get_ps_context
362 if _is_role_worker():
378 if _is_role_worker():
Dthor_layer.py28 from mindspore.parallel._ps_context import _is_role_worker, _get_ps_context
892 if _is_role_worker():
906 if _is_role_worker():
/third_party/mindspore/mindspore/parallel/
D_ps_context.py189 def _is_role_worker(): function