Searched refs:set_full_batch (Results 1 – 4 of 4) sorted by relevance
249 def set_full_batch(self, full_batch): member in _AutoParallelContext257 self._context_handle.set_full_batch(full_batch)278 self._context_handle.set_full_batch(dataset_strategy == "full_batch")644 "full_batch": auto_parallel_context().set_full_batch,
65 void set_full_batch(bool full_batch);
92 void ParallelContext::set_full_batch(bool full_batch) { full_batch_ = full_batch; } in set_full_batch() function in mindspore::parallel::ParallelContext
181 ….def("set_full_batch", &ParallelContext::set_full_batch, "Set whether load full batch on each devi… in PYBIND11_MODULE()