Home
last modified time | relevance | path

Searched refs:TestFxDetectInputWeightEqualization (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/pytorch/test/
H A Dtest_quantization.py118 … from quantization.fx.test_model_report_fx import TestFxDetectInputWeightEqualization # noqa: F401
/aosp_15_r20/external/pytorch/test/quantization/fx/
H A Dtest_model_report_fx.py1255 class TestFxDetectInputWeightEqualization(QuantizationTestCase): class
1271 self.block1 = TestFxDetectInputWeightEqualization.SimpleConv((3, 32))
1272 self.block2 = TestFxDetectInputWeightEqualization.SimpleConv((3, 3))