Home
last modified time | relevance | path

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

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