Searched defs:n_microbatches (Results 1 – 1 of 1) sorted by relevance
53 def _prepare_forward_infra(self, n_microbatches): argument56 def _prepare_backward_infra(self, n_microbatches): argument