Searched defs:_pre_forward (Results 1 – 3 of 3) sorted by relevance
216 def _pre_forward( member in FSDPState
344 def _pre_forward( function
1493 def _pre_forward(self, *inputs, **kwargs): member in DistributedDataParallel