Home
last modified time | relevance | path

Searched refs:LinearReluLinearModel (Results 1 – 4 of 4) sorted by relevance

/external/pytorch/test/quantization/fx/
Dtest_equalize_fx.py36 LinearReluLinearModel,
264 (LinearReluLinearModel, two_nn_layer_node_occurrence),
352 (LinearReluModel, 2), (LinearReluLinearModel, 2), (LinearReluAddModel, 2),
768 (LinearReluLinearModel, linearReluLinear_node_list),
Dtest_quantize_fx.py160 LinearReluLinearModel,
651 (LinearReluLinearModel, linearReluLinear_node_list)]
/external/pytorch/test/quantization/eager/
Dtest_quantize_eager_ptq.py49 LinearReluLinearModel,
1316 model = LinearReluLinearModel().eval()
/external/pytorch/torch/testing/_internal/
Dcommon_quantization.py1667 class LinearReluLinearModel(torch.nn.Module): class