Searched defs:pre_forward (Results 1 – 4 of 4) sorted by relevance
41 def pre_forward(module, module_args, module_kwargs): function
30 def pre_forward(self, module, args, kwargs): member in MyModule
245 def pre_forward(module, module_inputs): function
309 def pre_forward( member in FSDPParamGroup