Home
last modified time | relevance | path

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

/aosp_15_r20/external/executorch/examples/models/toy_model/
H A D__init__.py7 from .model import AddModule, AddMulModule, LinearModule, MulModule, SoftmaxModule
11 AddMulModule,
H A Dmodel.py57 class AddMulModule(torch.nn.Module, EagerModelBase): class
/aosp_15_r20/external/executorch/exir/backend/test/demos/
H A Dtest_delegate_aten_mode.py27 class AddMulModule(torch.nn.Module): class
36 add_mul_module = AddMulModule()
/aosp_15_r20/external/executorch/exir/backend/test/
H A Dtest_backends.py220 class AddMulModule(torch.nn.Module): class
229 add_mul_module = AddMulModule()
H A Dtest_backends_lifted.py240 class AddMulModule(torch.nn.Module): class
249 add_mul_module = AddMulModule()
/aosp_15_r20/external/executorch/exir/emit/test/
H A Dtest_emit.py1381 class AddMulModule(torch.nn.Module): class
1391 model = AddMulModule()