Lines Matching refs:MyModule
148 class MyModule(torch.nn.Module): class
161 model = MyModule().eval()
180 class MyModule(torch.nn.Module): class
200 ep = export(MyModule(), inp)
211 class MyModule(torch.nn.Module): class
223 ep = export(MyModule(), inp)
233 class MyModule(torch.nn.Module): class
247 MyModule(),
303 class MyModule(torch.nn.Module): class
311 exported_module = export(MyModule(), (input,)).run_decompositions()
345 class MyModule(torch.nn.Module): class
353 MyModule(),
689 class MyModule(torch.nn.Module): class
707 self.check_graph(MyModule(), inputs)
710 class MyModule(torch.nn.Module): class
721 self.check_graph(MyModule(), inputs)
856 class MyModule(torch.nn.Module): class
866 self.check_graph(MyModule(), inputs)
959 class MyModule(torch.nn.Module): class
972 model = MyModule().eval().cuda()
976 class MyModule(torch.nn.Module): class
987 m = MyModule()
992 class MyModule(torch.nn.Module): class
1002 m = MyModule()
1007 class MyModule(torch.nn.Module): class
1018 m = MyModule()