Lines Matching refs:BasicModule

53 class BasicModule(torch.nn.Module):  class
137         self.module = BasicModule()
138         self.scripted_module = torch.jit.script(BasicModule())
191         self.module = BasicModule()
192         self.scripted_module = torch.jit.script(BasicModule())
249         self.module = NestedModuleTest.NestedModule(BasicModule())
252             NestedModuleTest.NestedModule(BasicModule())
258             torch.jit.script(BasicModule()),
331             MiddleModule(BasicModule()),
332             MiddleModule(BasicModule()),
333             MiddleModule(BasicModule()),
337                 MiddleModule(BasicModule()),
338                 MiddleModule(BasicModule()),
339                 MiddleModule(BasicModule()),
344                 MiddleModule(BasicModule()),
345                 MiddleModule(BasicModule()),
346                 MiddleModule(BasicModule()),
426         mod = MiddleModule(BasicModule())
439             MiddleModule(BasicModule()),
440             MiddleModule(BasicModule()),
441             MiddleModule(BasicModule()),
798         self.module = BasicModule()
799         self.scripted_module = torch.jit.script(BasicModule())