Searched refs:LinearReluLinearModel (Results 1 – 4 of 4) sorted by relevance
| /aosp_15_r20/external/pytorch/test/quantization/fx/ |
| H A D | test_equalize_fx.py | 36 LinearReluLinearModel,
264 (LinearReluLinearModel, two_nn_layer_node_occurrence),
352 (LinearReluModel, 2), (LinearReluLinearModel, 2), (LinearReluAddModel, 2),
768 (LinearReluLinearModel, linearReluLinear_node_list),
|
| H A D | test_quantize_fx.py | 160 LinearReluLinearModel,
651 (LinearReluLinearModel, linearReluLinear_node_list)]
|
| /aosp_15_r20/external/pytorch/test/quantization/eager/ |
| H A D | test_quantize_eager_ptq.py | 49 LinearReluLinearModel,
1316 model = LinearReluLinearModel().eval()
|
| /aosp_15_r20/external/pytorch/torch/testing/_internal/ |
| H A D | common_quantization.py | 1667 class LinearReluLinearModel(torch.nn.Module): class
|