Home
last modified time | relevance | path

Searched refs:SingleLayerLinearModel (Results 1 – 9 of 9) sorted by relevance

/aosp_15_r20/external/pytorch/test/quantization/core/
H A Dtest_docs.py12 SingleLayerLinearModel,
129 input_fp32 = SingleLayerLinearModel().get_example_inputs()
130 global_inputs = {"UserModel": SingleLayerLinearModel, "input_fp32": input_fp32}
H A Dtest_workflow_module.py51 SingleLayerLinearModel,
394 model = SingleLayerLinearModel().eval()
/aosp_15_r20/external/pytorch/test/quantization/fx/
H A Dtest_equalize_fx.py24 SingleLayerLinearModel,
259 tests = [(SingleLayerLinearModel, single_nn_layer_node_occurrence),
349 tests = [(SingleLayerLinearModel, 2), (LinearAddModel, 2), (TwoLayerLinearModel, 2),
422 tests = [SingleLayerLinearModel, TwoLayerLinearModel,
474 tests = [SingleLayerLinearModel, TwoLayerLinearModel,
532 tests = [SingleLayerLinearModel, TwoLayerLinearModel, SingleLayerFunctionalLinearModel]
761 tests = [(SingleLayerLinearModel, linear_node_list),
798 tests = [SingleLayerLinearModel, TwoLayerLinearModel, LinearAddModel,
H A Dtest_model_report_fx.py27 SingleLayerLinearModel,
511 self.mod1 = SingleLayerLinearModel()
642 self.mod1 = SingleLayerLinearModel()
695 self.fc1 = SingleLayerLinearModel()
H A Dtest_numeric_suite_fx.py33 SingleLayerLinearModel,
2774 (SingleLayerLinearModel(), None),
2811 (SingleLayerLinearModel(), None),
2848 (SingleLayerLinearModel(), None),
/aosp_15_r20/external/pytorch/test/ao/sparsity/
H A Dtest_sparsity_utils.py18 SingleLayerLinearModel,
30 SingleLayerLinearModel,
/aosp_15_r20/external/pytorch/test/quantization/eager/
H A Dtest_fuse_eager.py31 SingleLayerLinearModel,
368 dummy_model = SingleLayerLinearModel().train()
/aosp_15_r20/external/pytorch/test/quantization/jit/
H A Dtest_quantize_jit.py64 SingleLayerLinearModel,
3564 linear_model = SingleLayerLinearModel().eval()
3608 linear_model = SingleLayerLinearModel().eval()
3831 linear_model = SingleLayerLinearModel().eval()
3862 linear_model = SingleLayerLinearModel().eval()
/aosp_15_r20/external/pytorch/torch/testing/_internal/
H A Dcommon_quantization.py1352 class SingleLayerLinearModel(torch.nn.Module): class