Searched refs:LinearReluModel (Results 1 – 4 of 4) sorted by relevance
| /aosp_15_r20/external/pytorch/test/quantization/fx/ |
| H A D | test_equalize_fx.py | 35 LinearReluModel,
263 (LinearReluModel, single_nn_layer_node_occurrence),
352 (LinearReluModel, 2), (LinearReluLinearModel, 2), (LinearReluAddModel, 2),
767 (LinearReluModel, linearRelu_node_list),
|
| H A D | test_quantize_fx.py | 161 LinearReluModel,
650 tests = [(LinearReluModel, linearRelu_node_list),
|
| /aosp_15_r20/external/pytorch/torch/testing/_internal/ |
| H A D | common_quantization.py | 1653 class LinearReluModel(torch.nn.Module): class
1867 self.sub1 = LinearReluModel()
1880 self.sub1 = LinearReluModel()
1899 self.sub1 = LinearReluModel()
1914 self.sub1 = LinearReluModel()
1940 self.sub1 = LinearReluModel()
2931 class LinearReluModel(torch.nn.Module): class in TestHelperModules
|
| /aosp_15_r20/external/pytorch/test/quantization/pt2e/ |
| H A D | test_xnnpack_quantizer.py | 160 m_eager = TestHelperModules.LinearReluModel().eval()
|