Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/test/jit/
H A 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
H A Dtest_tracer.py2788 class WrapperModule(torch.nn.Module): class
/aosp_15_r20/external/pytorch/benchmarks/framework_overhead_benchmark/
H A Dpt_wrapper_module.py4 class WrapperModule: class
/aosp_15_r20/external/executorch/exir/capture/
H A D_capture.py158 class WrapperModule(torch.nn.Module): class
/aosp_15_r20/external/executorch/test/end2end/
H A Dexported_module.py160 class WrapperModule(nn.Module): class
/aosp_15_r20/external/executorch/test/models/
H A Dexport_delegated_program.py90 class WrapperModule(torch.nn.Module): class
/aosp_15_r20/external/pytorch/test/inductor/
H A Dtest_aot_inductor_utils.py16 class WrapperModule(torch.nn.Module): class
/aosp_15_r20/external/executorch/exir/tests/
H A Dtest_delegate.py29 class WrapperModule(torch.nn.Module): class
H A Dtest_verification.py22 class WrapperModule(torch.nn.Module): class
/aosp_15_r20/external/pytorch/torch/_dynamo/backends/
H A Ddistributed.py147 class WrapperModule(torch.nn.Module): class
/aosp_15_r20/external/pytorch/test/fx/
H A Dtest_matcher_utils.py25 class WrapperModule(torch.nn.Module): class
/aosp_15_r20/external/pytorch/test/dynamo/
H A Dtest_aot_autograd.py717 class WrapperModule(torch.nn.Module): class
732 class WrapperModule(torch.nn.Module): class
H A Dtest_functions.py3366 class WrapperModule(torch.nn.Module): class
/aosp_15_r20/external/executorch/extension/pybindings/test/
H A Dmake_test.py103 class WrapperModule(torch.nn.Module): class
/aosp_15_r20/external/executorch/exir/program/test/
H A Dtest_program.py120 class WrapperModule(torch.nn.Module): class
/aosp_15_r20/external/executorch/exir/emit/test/
H A Dtest_emit.py66 class WrapperModule(torch.nn.Module): class
/aosp_15_r20/external/pytorch/torch/testing/_internal/distributed/rpc/
H A Ddist_autograd_test.py2640 class WrapperModule(nn.Module): class
/aosp_15_r20/external/pytorch/test/
H A Dtest_fx.py829 class WrapperModule(torch.nn.Module): class