Home
last modified time | relevance | path

Searched refs:ModuleAdd (Results 1 – 10 of 10) sorted by relevance

/aosp_15_r20/external/executorch/extension/pybindings/test/
H A Dmake_test.py19 class ModuleAdd(torch.nn.Module): class
23 super(ModuleAdd, self).__init__()
144 exported_program, inputs = create_program(ModuleAdd())
180 exported_program, inputs = create_program(ModuleAdd())
230 exported_program, inputs = create_program(ModuleAdd())
245 eager_module = ModuleAdd()
310 exported_program, inputs = create_program(ModuleAdd())
356 exported_program, inputs = create_program(ModuleAdd())
366 exported_program, inputs = create_program(ModuleAdd())
/aosp_15_r20/external/executorch/runtime/test/
H A Dtest_runtime.py15 ModuleAdd,
23 ep, inputs = create_program(ModuleAdd())
48 ep, inputs = create_program(ModuleAdd())
57 ep, inputs = create_program(ModuleAdd())
/aosp_15_r20/external/executorch/test/models/
H A Dtargets.bzl62 "ModuleAdd",
73 # To use one, depend on a target like ":exported_programs[ModuleAdd.pte]".
134 # ":exported_delegated_programs[ModuleAdd.pte]" or
135 # ":exported_delegated_programs[ModuleAdd-nosegments.pte]" (which does not
H A Dexport_program.py80 class ModuleAdd(nn.Module): class
82 super(ModuleAdd, self).__init__()
/aosp_15_r20/external/executorch/backends/apple/mps/test/
H A Dtest_mps_models.py63 class ModuleAdd(torch.nn.Module): class
65 super(ModuleAdd, self).__init__()
368 "module_add": lambda: (ModuleAdd(), ModuleAdd().get_random_inputs()),
/aosp_15_r20/external/pytorch/test/jit/
H A Dtest_backends.py736 class ModuleAdd(torch.nn.Module): class in CompModuleTestSameNameWithCompiler
758 torch.jit.script(ModuleAdd()), # noqa: F821
763 torch.jit.script(ModuleAdd()), # noqa: F821
/aosp_15_r20/external/executorch/test/end2end/
H A Dtest_end2end.py115 class ModuleAdd(nn.Module): class
117 super(ModuleAdd, self).__init__()
621 maketest(ModuleAdd)(self)
/aosp_15_r20/external/executorch/extension/runner_util/test/
H A Dtargets.bzl30 …ET_MODULE_ADD_PATH": "$(location fbcode//executorch/test/models:exported_programs[ModuleAdd.pte])",
/aosp_15_r20/external/executorch/extension/training/module/test/
H A Dtargets.bzl18 …ET_MODULE_ADD_PATH": "$(location fbcode//executorch/test/models:exported_programs[ModuleAdd.pte])",
/aosp_15_r20/external/executorch/runtime/executor/test/
H A Dtargets.bzl106 …ET_MODULE_ADD_PATH": "$(location fbcode//executorch/test/models:exported_programs[ModuleAdd.pte])",