Home
last modified time | relevance | path

Searched defs:WrapperModule (Results 1 – 18 of 18) sorted by relevance

/external/pytorch/benchmarks/framework_overhead_benchmark/
Dpt_wrapper_module.py4 class WrapperModule: class
/external/executorch/exir/capture/
D_capture.py158 class WrapperModule(torch.nn.Module): class
/external/executorch/test/end2end/
Dexported_module.py160 class WrapperModule(nn.Module): class
/external/executorch/test/models/
Dexport_delegated_program.py90 class WrapperModule(torch.nn.Module): class
/external/pytorch/test/inductor/
Dtest_aot_inductor_utils.py16 class WrapperModule(torch.nn.Module): class
/external/executorch/exir/tests/
Dtest_delegate.py29 class WrapperModule(torch.nn.Module): class
Dtest_verification.py22 class WrapperModule(torch.nn.Module): class
/external/pytorch/torch/_dynamo/backends/
Ddistributed.py147 class WrapperModule(torch.nn.Module): class
/external/pytorch/test/fx/
Dtest_matcher_utils.py25 class WrapperModule(torch.nn.Module): class
/external/executorch/extension/pybindings/test/
Dmake_test.py103 class WrapperModule(torch.nn.Module): class
/external/pytorch/test/jit/
Dtest_freezing.py1590 class WrapperModule(torch.nn.Module): class
1623 class WrapperModule(torch.nn.Module): class
1677 class WrapperModule(torch.nn.Module): class
1743 class WrapperModule(torch.nn.Module): class
1869 class WrapperModule(torch.nn.Module): class
1899 class WrapperModule(torch.nn.Module): class
1946 class WrapperModule(torch.nn.Module): class
Dtest_tracer.py2788 class WrapperModule(torch.nn.Module): class
/external/pytorch/test/dynamo/
Dtest_aot_autograd.py717 class WrapperModule(torch.nn.Module): class
732 class WrapperModule(torch.nn.Module): class
Dtest_functions.py3366 class WrapperModule(torch.nn.Module): class
/external/executorch/exir/program/test/
Dtest_program.py120 class WrapperModule(torch.nn.Module): class
/external/executorch/exir/emit/test/
Dtest_emit.py66 class WrapperModule(torch.nn.Module): class
/external/pytorch/torch/testing/_internal/distributed/rpc/
Ddist_autograd_test.py2640 class WrapperModule(nn.Module): class
/external/pytorch/test/
Dtest_fx.py829 class WrapperModule(torch.nn.Module): class