Searched refs:OptimizedModule (Results 1 – 7 of 7) sorted by relevance
| /external/pytorch/torch/_dynamo/ |
| D | __init__.py | 28 OptimizedModule,
|
| D | testing.py | 65 if isinstance(model, eval_frame.OptimizedModule):
78 if isinstance(model, eval_frame.OptimizedModule):
|
| D | eval_frame.py | 142 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)
|
| /external/pytorch/test/dynamo/ |
| D | test_modules.py | 1685 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)
|
| /external/pytorch/torch/fx/experimental/ |
| D | proxy_tensor.py | 1688 from torch._dynamo import OptimizedModule
1697 if isinstance(m, (OptimizedModule, GraphModule)):
|
| /external/pytorch/torch/distributed/checkpoint/ |
| D | state_dict.py | 213 elif isinstance(curr_obj, torch._dynamo.eval_frame.OptimizedModule):
|
| /external/pytorch/torch/_dynamo/variables/ |
| D | builder.py | 1318 from ..eval_frame import OptimizedModule
1322 if istype(value, OptimizedModule):
|