Searched refs:set_grad_accumulation_step (Results 1 – 4 of 4) sorted by relevance
301 def set_grad_accumulation_step(self, grad_accumulation_step): member in _AutoParallelContext310 self._context_handle.set_grad_accumulation_step(grad_accumulation_step)647 "grad_accumulation_step": auto_parallel_context().set_grad_accumulation_step,
86 void set_grad_accumulation_step(int64_t grad_accumulation_step);
98 void ParallelContext::set_grad_accumulation_step(int64_t grad_accumulation_step) { in set_grad_accumulation_step() function in mindspore::parallel::ParallelContext
152 ….def("set_grad_accumulation_step", &ParallelContext::set_grad_accumulation_step, "Set grad accumul… in PYBIND11_MODULE()