Home
last modified time | relevance | path

Searched refs:ModuleSimpleTrain (Results 1 – 6 of 6) sorted by relevance

/aosp_15_r20/external/executorch/extension/training/pybindings/test/
H A Dtest.py22 class ModuleSimpleTrain(torch.nn.Module): class in TestTraining
35 m = self.ModuleSimpleTrain()
/aosp_15_r20/external/executorch/extension/training/module/test/
H A Dtargets.bzl19 …TRAIN_PATH": "$(location fbcode//executorch/test/models:exported_programs[ModuleSimpleTrain.pte])",
/aosp_15_r20/external/executorch/extension/training/test/
H A Dtargets.bzl18 …TRAIN_PATH": "$(location fbcode//executorch/test/models:exported_programs[ModuleSimpleTrain.pte])",
/aosp_15_r20/external/executorch/test/models/
H A Dtargets.bzl69 "ModuleSimpleTrain",
H A Dexport_program.py168 class ModuleSimpleTrain(torch.nn.Module): class
/aosp_15_r20/external/executorch/runtime/executor/test/
H A Dtargets.bzl111 …TRAIN_PATH": "$(location fbcode//executorch/test/models:exported_programs[ModuleSimpleTrain.pte])",