Home
last modified time | relevance | path

Searched refs:WrappedModule (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/executorch/exir/backend/test/
H A Dtest_lowered_backend_module.py51 class WrappedModule(torch.nn.Module): class
61 export(WrappedModule(), example_inputs),
199 class WrappedModule(torch.nn.Module): class
207 wrapped_module = WrappedModule(lowered_module)
/aosp_15_r20/external/executorch/backends/xnnpack/test/
H A Dtest_xnnpack_utils.py182 class WrappedModule(torch.nn.Module): class
190 edge_program = capture_graph_for_xnnpack(WrappedModule(), sample_inputs)