Searched refs:LinearAddModel (Results 1 – 4 of 4) sorted by relevance
| /aosp_15_r20/external/pytorch/test/quantization/jit/ |
| H A D | test_ondevice_quantization.py | 19 LinearAddModel,
169 model = LinearAddModel()
246 model = LinearAddModel()
252 model = LinearAddModel()
509 model = LinearAddModel()
515 model = LinearAddModel()
521 model = LinearAddModel()
|
| /aosp_15_r20/external/pytorch/test/ao/sparsity/ |
| H A D | test_sparsity_utils.py | 16 LinearAddModel,
32 LinearAddModel,
|
| /aosp_15_r20/external/pytorch/test/quantization/fx/ |
| H A D | test_equalize_fx.py | 26 LinearAddModel,
349 tests = [(SingleLayerLinearModel, 2), (LinearAddModel, 2), (TwoLayerLinearModel, 2),
762 (LinearAddModel, linearAdd_node_list),
798 tests = [SingleLayerLinearModel, TwoLayerLinearModel, LinearAddModel,
|
| /aosp_15_r20/external/pytorch/torch/testing/_internal/ |
| H A D | common_quantization.py | 1390 class LinearAddModel(nn.Module): class
|