Lines Matching refs:CustomModuleInterface

260         class CustomModuleInterface(torch.nn.Module):  class
263         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):  class
334         class CustomModuleList(CustomModuleInterface, torch.nn.ModuleList):
336                 CustomModuleInterface.__init__(self)
432         class CustomModuleInterface(torch.nn.Module):  class
435         class CustomModuleDict(CustomModuleInterface, torch.nn.ModuleDict):
437                 CustomModuleInterface.__init__(self)