Home
last modified time | relevance | path

Searched refs:AnnotatedSingleLayerLinearModel (Results 1 – 6 of 6) sorted by relevance

/aosp_15_r20/external/pytorch/test/quantization/eager/
H A Dtest_numeric_suite_eager.py30 AnnotatedSingleLayerLinearModel,
132 model_list = [AnnotatedSingleLayerLinearModel(qengine)]
231 model_list = [AnnotatedSingleLayerLinearModel(qengine)]
407 model_list = [AnnotatedSingleLayerLinearModel(qengine)]
H A Dtest_quantize_eager_ptq.py31 AnnotatedSingleLayerLinearModel,
345 model = AnnotatedSingleLayerLinearModel(qengine)
367 base = AnnotatedSingleLayerLinearModel(qengine)
376 model = AnnotatedSingleLayerLinearModel(qengine)
/aosp_15_r20/external/pytorch/test/mobile/
H A Dtest_lite_script_module.py14 AnnotatedSingleLayerLinearModel,
507 model_class=AnnotatedSingleLayerLinearModel, qengine="qnnpack"
/aosp_15_r20/external/pytorch/test/quantization/core/
H A Dtest_workflow_module.py49 AnnotatedSingleLayerLinearModel,
621 model = AnnotatedSingleLayerLinearModel()
/aosp_15_r20/external/pytorch/test/quantization/jit/
H A Dtest_quantize_jit.py55 AnnotatedSingleLayerLinearModel,
3561 annotated_linear_model = AnnotatedSingleLayerLinearModel(
3597 annotated_linear_model = AnnotatedSingleLayerLinearModel("fbgemm").eval()
3830 annotated_linear_model = AnnotatedSingleLayerLinearModel("qnnpack").eval()
/aosp_15_r20/external/pytorch/torch/testing/_internal/
H A Dcommon_quantization.py1364 class AnnotatedSingleLayerLinearModel(torch.nn.Module): class