Searched defs:forward_prefetch (Results 1 – 3 of 3) sorted by relevance
145 forward_prefetch, argument214 forward_prefetch, argument
392 forward_prefetch, argument
1137 def _check_forward_prefetch(self, fsdp_model, forward_prefetch): argument