Searched refs:OneTwoModule (Results 1 – 2 of 2) sorted by relevance
68 class OneTwoModule(nn.Module): class110 make_global(OneTwoModule, OneTwoClass)112 def use_module_interface(mod_list: List[OneTwoModule], x: torch.Tensor): argument136 mod_interface: OneTwoModule, x: Tensor argument182 class OneTwoModule(nn.Module): class192 make_global(OneTwoModule)195 def as_module_interface(x: OneTwoModule) -> OneTwoModule: argument
246 class OneTwoModule(nn.Module): class276 make_global(OneTwoModule)279 sub: OneTwoModule288 def use_module_interface(mod_list: List[OneTwoModule], x: torch.Tensor): argument