Searched refs:set_optimizer_weight_shard_size (Results 1 – 4 of 4) sorted by relevance
549 def set_optimizer_weight_shard_size(self, optimizer_weight_shard_size): member in _AutoParallelContext565 self._context_handle.set_optimizer_weight_shard_size(optimizer_weight_shard_size)650 "optimizer_weight_shard_size": auto_parallel_context().set_optimizer_weight_shard_size,
102 void set_optimizer_weight_shard_size(int64_t optimizer_weight_shard_size);
145 void ParallelContext::set_optimizer_weight_shard_size(int64_t optimizer_weight_shard_size) { in set_optimizer_weight_shard_size() function in mindspore::parallel::ParallelContext
191 .def("set_optimizer_weight_shard_size", &ParallelContext::set_optimizer_weight_shard_size, in PYBIND11_MODULE()