Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/test/
H A Dtest_model_dump.py225 class ComposedModule(torch.nn.Module): class
236 torch.jit.script(ComposedModule()).eval(),
/aosp_15_r20/external/executorch/docs/source/tutorials_source/
H A Dexport-to-executorch-tutorial.py412 class ComposedModule(torch.nn.Module): class
426 aten_dialect: ExportedProgram = export(ComposedModule(), example_args)