Searched defs:fw_pre_hook (Results 1 – 5 of 5) sorted by relevance
729 def fw_pre_hook(h_module, input): function1454 def fw_pre_hook(h_module, input): function
392 def fw_pre_hook(fused_module_class, h_module, input): function
525 def fw_pre_hook(h_module, input): function
28 def fw_pre_hook(mod, inp): function
514 def fw_pre_hook(module: nn.Module, _inputs): function