Searched refs:LinearModule (Results 1 – 6 of 6) sorted by relevance
| /external/executorch/examples/models/toy_model/ |
| D | __init__.py | 7 from .model import AddModule, AddMulModule, LinearModule, MulModule, SoftmaxModule
12 LinearModule,
|
| D | model.py | 27 class LinearModule(torch.nn.Module, EagerModelBase): class
|
| /external/executorch/backends/xnnpack/test/ |
| D | test_xnnpack_utils.py | 520 self, LinearModule, example_inputs argument
522 linear = LinearModule()
|
| /external/executorch/backends/vulkan/test/ |
| D | test_vulkan_delegate.py | 621 class LinearModule(torch.nn.Module): class
629 module = LinearModule()
|
| /external/pytorch/test/quantization/fx/ |
| D | test_quantize_fx.py | 1072 class LinearModule(torch.nn.Module): class
1151 LinearModule,
1159 LinearModule,
1763 class LinearModule(torch.nn.Module): class
1777 (LinearModule, (), (linear_module_input,),
|
| /external/pytorch/test/ |
| D | test_autograd.py | 7230 class LinearModule(torch.nn.Module): class
7242 model = LinearModule()
|