Searched defs:module_inputs (Results 1 – 2 of 2) sorted by relevance
245 def pre_forward(module, module_inputs): argument
539 def module_inputs(self) -> Sequence[torch.fx.Node]: member in _ModuleNode