Lines Matching refs:MyModule
81 class MyModule(torch.nn.Module): class
96 orig_eager = MyModule()
124 class MyModule(torch.nn.Module): class
136 eager_module = MyModule()
159 class MyModule(torch.nn.Module): class
172 eager_module = MyModule()
231 class MyModule(torch.nn.Module): class
242 orig_eager = MyModule()
391 class MyModule(torch.nn.Module): class
406 orig_eager = MyModule()
420 class MyModule(torch.nn.Module): class
429 orig_eager = MyModule()
454 class MyModule(torch.nn.Module): class
462 orig_eager = MyModule()
513 class MyModule(torch.nn.Module): class
521 orig_eager = MyModule()
763 class MyModule(torch.nn.Module): class
781 mod = MyModule()
925 class MyModule(torch.nn.Module): class
940 orig_eager = MyModule()