Home
last modified time | relevance | path

Searched refs:OptimizedModule (Results 1 – 7 of 7) sorted by relevance

/aosp_15_r20/external/pytorch/torch/_dynamo/
H A Dtesting.py65 if isinstance(model, eval_frame.OptimizedModule):
78 if isinstance(model, eval_frame.OptimizedModule):
H A Deval_frame.py142 class OptimizedModule(torch.nn.Module): class
227 if name in OptimizedModule._opt_mod_attributes:
381 new_mod = OptimizedModule(mod, self)
606 new_mod = OptimizedModule(mod, self)
H A D__init__.py28 OptimizedModule,
/aosp_15_r20/external/pytorch/test/dynamo/
H A Dtest_modules.py1685 self.assertIsInstance(opt_mod, torch._dynamo.OptimizedModule)
1748 self.assertIsInstance(opt_mod, torch._dynamo.OptimizedModule)
2119 self.assertIsInstance(opt_outer_mod, torch._dynamo.OptimizedModule)
2148 self.assertIsInstance(opt_outer_mod, torch._dynamo.OptimizedModule)
/aosp_15_r20/external/pytorch/torch/fx/experimental/
H A Dproxy_tensor.py1688 from torch._dynamo import OptimizedModule
1697 if isinstance(m, (OptimizedModule, GraphModule)):
/aosp_15_r20/external/pytorch/torch/distributed/checkpoint/
H A Dstate_dict.py213 elif isinstance(curr_obj, torch._dynamo.eval_frame.OptimizedModule):
/aosp_15_r20/external/pytorch/torch/_dynamo/variables/
H A Dbuilder.py1318 from ..eval_frame import OptimizedModule
1322 if istype(value, OptimizedModule):