Home
last modified time | relevance | path

Searched refs:TwoLayerLinearModel (Results 1 – 8 of 8) sorted by relevance

/aosp_15_r20/external/pytorch/test/quantization/fx/
H A Dtest_equalize_fx.py25 TwoLayerLinearModel,
260 (TwoLayerLinearModel, two_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]
763 (TwoLayerLinearModel, linear2_node_list),
798 tests = [SingleLayerLinearModel, TwoLayerLinearModel, LinearAddModel,
H A Dtest_model_report_fx.py28 TwoLayerLinearModel,
187 TwoLayerLinearModel(),
189 TwoLayerLinearModel().get_example_inputs()[0],
H A Dtest_numeric_suite_fx.py37 TwoLayerLinearModel
2399 m = TwoLayerLinearModel().eval()
2427 m = TwoLayerLinearModel().eval()
2456 m = TwoLayerLinearModel().eval()
/aosp_15_r20/external/pytorch/test/ao/sparsity/
H A Dtest_sparsity_utils.py19 TwoLayerLinearModel,
31 TwoLayerLinearModel,
/aosp_15_r20/external/pytorch/test/quantization/eager/
H A Dtest_quantize_eager_ptq.py35 TwoLayerLinearModel,
661 model = TwoLayerLinearModel()
675 model = TwoLayerLinearModel()
1132 model = TwoLayerLinearModel().eval()
1150 model = quantize_dynamic(TwoLayerLinearModel().eval(), qconfig_dict)
1154 model = quantize_dynamic(TwoLayerLinearModel().eval(), {'fc2'}, dtype=dtype)
H A Dtest_quantize_eager_qat.py51 TwoLayerLinearModel,
470 model = TwoLayerLinearModel()
486 model = TwoLayerLinearModel()
502 model = TwoLayerLinearModel()
/aosp_15_r20/external/pytorch/test/mobile/
H A Dtest_lite_script_module.py16 TwoLayerLinearModel,
513 quantized_model = self._create_quantized_model(model_class=TwoLayerLinearModel)
/aosp_15_r20/external/pytorch/torch/testing/_internal/
H A Dcommon_quantization.py1594 class TwoLayerLinearModel(torch.nn.Module): class
1611 self.subm = TwoLayerLinearModel()
1868 self.sub2 = TwoLayerLinearModel()
1881 self.sub2 = TwoLayerLinearModel()
1900 self.sub2 = QuantWrapper(TwoLayerLinearModel())
1915 self.sub2 = TwoLayerLinearModel()
1941 self.sub2 = QuantWrapper(TwoLayerLinearModel())