Home
last modified time | relevance | path

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

/third_party/mindspore/mindspore/parallel/
D_auto_parallel_context.py82 def set_global_rank(self, global_rank): member in _AutoParallelContext
95 self._context_handle.set_global_rank(global_rank)
633 "global_rank": auto_parallel_context().set_global_rank,
/third_party/mindspore/tests/ut/python/parallel/
Dtest_set_auto_parallel_context.py47 auto_parallel_context().set_global_rank(4)
/third_party/mindspore/mindspore/ccsrc/frontend/parallel/
Dcontext.h83 void set_global_rank(int64_t global_rank);
Dcontext.cc85 void ParallelContext::set_global_rank(int64_t global_rank) { in set_global_rank() function in mindspore::parallel::ParallelContext
/third_party/mindspore/mindspore/ccsrc/pipeline/jit/
Dinit.cc141 .def("set_global_rank", &ParallelContext::set_global_rank, "Set global rank.") in PYBIND11_MODULE()