Searched refs:get_full_batch (Results 1 – 2 of 2) sorted by relevance
259 def get_full_batch(self): member in _AutoParallelContext264 return self._context_handle.get_full_batch()296 if self._context_handle.get_full_batch():668 "full_batch": auto_parallel_context().get_full_batch,
40 return auto_parallel_context().get_full_batch()