Searched refs:CustomModuleInterface (Results 1 – 1 of 1) sorted by relevance
260 class CustomModuleInterface(torch.nn.Module): class263 class CustomModuleList(CustomModuleInterface, torch.nn.ModuleList):265 CustomModuleInterface.__init__(self)268 class CustomSequential(CustomModuleInterface, torch.nn.Sequential):270 CustomModuleInterface.__init__(self)273 class CustomModuleDict(CustomModuleInterface, torch.nn.ModuleDict):275 CustomModuleInterface.__init__(self)331 class CustomModuleInterface(torch.nn.Module): class334 class CustomModuleList(CustomModuleInterface, torch.nn.ModuleList):336 CustomModuleInterface.__init__(self)[all …]