Searched defs:OneTwoModule (Results 1 – 2 of 2) sorted by relevance
68 class OneTwoModule(nn.Module): class112 def use_module_interface(mod_list: List[OneTwoModule], x: torch.Tensor):136 mod_interface: OneTwoModule, x: Tensor182 class OneTwoModule(nn.Module): class195 def as_module_interface(x: OneTwoModule) -> OneTwoModule:
246 class OneTwoModule(nn.Module): class288 def use_module_interface(mod_list: List[OneTwoModule], x: torch.Tensor):