Searched defs:backward_prefetch (Results 1 – 3 of 3) sorted by relevance
81 def _dist_train(self, backward_prefetch=BackwardPrefetch.BACKWARD_PRE): argument
391 backward_prefetch, argument
1134 def _check_backward_prefetch(self, fsdp_model, backward_prefetch): argument