| /aosp_15_r20/external/pytorch/test/fx/ |
| H A D | test_matcher_utils.py | 25 class WrapperModule(torch.nn.Module): class
176 pattern_gm = capture_pre_autograd_graph(WrapperModule(pattern), example_inputs)
207 pattern_gm = capture_pre_autograd_graph(WrapperModule(pattern), example_inputs)
210 WrapperModule(target_graph), example_inputs
|
| /aosp_15_r20/external/executorch/exir/tests/ |
| H A D | test_delegate.py | 29 class WrapperModule(torch.nn.Module): class
48 edge_ir_m = to_edge(export(WrapperModule(g), inputs))
58 WrapperModule(f),
|
| H A D | test_verification.py | 22 class WrapperModule(torch.nn.Module): class
38 to_edge(export(WrapperModule(f), (torch.randn(2),)))
|
| /aosp_15_r20/external/pytorch/benchmarks/framework_overhead_benchmark/ |
| H A D | framework_overhead_benchmark.py | 3 from pt_wrapper_module import WrapperModule
55 module = WrapperModule(module_type, module_config, args.debug, args.save)
|
| H A D | pt_wrapper_module.py | 4 class WrapperModule: class
|
| /aosp_15_r20/external/pytorch/test/inductor/ |
| H A D | test_aot_inductor_utils.py | 16 class WrapperModule(torch.nn.Module): class
35 model = WrapperModule(model)
|
| /aosp_15_r20/external/executorch/test/models/ |
| H A D | export_delegated_program.py | 90 class WrapperModule(torch.nn.Module): class
99 export(WrapperModule(getattr(eager_module, method)), args=inputs)
|
| /aosp_15_r20/external/pytorch/test/jit/ |
| H A D | test_freezing.py | 1590 class WrapperModule(torch.nn.Module): class
1600 m = torch.jit.script(WrapperModule())
1623 class WrapperModule(torch.nn.Module): class
1641 m = torch.jit.script(WrapperModule())
1677 class WrapperModule(torch.nn.Module): class
1687 m = WrapperModule()
1743 class WrapperModule(torch.nn.Module): class
1753 m = WrapperModule()
1869 class WrapperModule(torch.nn.Module): class
1879 m = WrapperModule()
[all …]
|
| H A D | test_tracer.py | 2788 class WrapperModule(torch.nn.Module): class
2802 self.sub2 = WrapperModule(m)
|
| /aosp_15_r20/external/executorch/exir/capture/ |
| H A D | _capture.py | 158 class WrapperModule(torch.nn.Module): class
215 mod = f if isinstance(f, torch.nn.Module) else WrapperModule(f)
|
| /aosp_15_r20/external/executorch/extension/pybindings/test/ |
| H A D | make_test.py | 103 class WrapperModule(torch.nn.Module): class
115 wrapped_mod = WrapperModule(getattr(eager_module, method_name))
|
| /aosp_15_r20/external/pytorch/torch/_dynamo/backends/ |
| H A D | distributed.py | 147 class WrapperModule(torch.nn.Module): class
180 wrapper = WrapperModule(
|
| /aosp_15_r20/external/pytorch/test/dynamo/ |
| H A D | test_aot_autograd.py | 717 class WrapperModule(torch.nn.Module): class
732 class WrapperModule(torch.nn.Module): class
741 return WrapperModule()
749 WrapperModule(split_gm.submod_0), example_inps
752 WrapperModule(split_gm.submod_1), submod_1_inps
|
| H A D | test_functions.py | 3366 class WrapperModule(torch.nn.Module): class
3425 mod = WrapperModule()
|
| /aosp_15_r20/external/executorch/test/end2end/ |
| H A D | exported_module.py | 160 class WrapperModule(nn.Module): class
|
| /aosp_15_r20/external/executorch/exir/program/test/ |
| H A D | test_program.py | 120 class WrapperModule(torch.nn.Module): class
550 callable = WrapperModule(callable)
|
| /aosp_15_r20/external/executorch/exir/emit/test/ |
| H A D | test_emit.py | 66 class WrapperModule(torch.nn.Module): class
1088 WrapperModule(fn),
|
| /aosp_15_r20/external/pytorch/torch/testing/_internal/distributed/rpc/ |
| H A D | dist_autograd_test.py | 2640 class WrapperModule(nn.Module): class
2758 WrapperModule,
|
| /aosp_15_r20/external/pytorch/test/ |
| H A D | test_fx.py | 829 class WrapperModule(torch.nn.Module): class
834 wrapper = WrapperModule(interpreter)
|