Searched refs:set_device_num (Results 1 – 5 of 5) sorted by relevance
62 def set_device_num(self, device_num): member in _AutoParallelContext75 self._context_handle.set_device_num(device_num)632 "device_num": auto_parallel_context().set_device_num,
41 auto_parallel_context().set_device_num(4)
77 void set_device_num(int64_t device_num);
80 void ParallelContext::set_device_num(int64_t device_num) { in set_device_num() function in mindspore::parallel::ParallelContext
138 .def("set_device_num", &ParallelContext::set_device_num, "Set device num.") in PYBIND11_MODULE()