Searched refs:SingleLayerLinearModel (Results 1 – 9 of 9) sorted by relevance
/aosp_15_r20/external/pytorch/test/quantization/core/ |
H A D | test_docs.py | 12 SingleLayerLinearModel, 129 input_fp32 = SingleLayerLinearModel().get_example_inputs() 130 global_inputs = {"UserModel": SingleLayerLinearModel, "input_fp32": input_fp32}
|
H A D | test_workflow_module.py | 51 SingleLayerLinearModel, 394 model = SingleLayerLinearModel().eval()
|
/aosp_15_r20/external/pytorch/test/quantization/fx/ |
H A D | test_equalize_fx.py | 24 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 D | test_model_report_fx.py | 27 SingleLayerLinearModel, 511 self.mod1 = SingleLayerLinearModel() 642 self.mod1 = SingleLayerLinearModel() 695 self.fc1 = SingleLayerLinearModel()
|
H A D | test_numeric_suite_fx.py | 33 SingleLayerLinearModel, 2774 (SingleLayerLinearModel(), None), 2811 (SingleLayerLinearModel(), None), 2848 (SingleLayerLinearModel(), None),
|
/aosp_15_r20/external/pytorch/test/ao/sparsity/ |
H A D | test_sparsity_utils.py | 18 SingleLayerLinearModel, 30 SingleLayerLinearModel,
|
/aosp_15_r20/external/pytorch/test/quantization/eager/ |
H A D | test_fuse_eager.py | 31 SingleLayerLinearModel, 368 dummy_model = SingleLayerLinearModel().train()
|
/aosp_15_r20/external/pytorch/test/quantization/jit/ |
H A D | test_quantize_jit.py | 64 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 D | common_quantization.py | 1352 class SingleLayerLinearModel(torch.nn.Module): class
|