Searched refs:_test_function_import (Results 1 – 4 of 4) sorted by relevance
| /aosp_15_r20/external/pytorch/test/quantization/ao_migration/ |
| H A D | test_ao_migration.py | 33 self._test_function_import("functional", function_list, base="nn.quantized")
69 self._test_function_import("modules", module_list, base="nn.quantized")
80 self._test_function_import(
89 self._test_function_import(
104 self._test_function_import("conv", function_list, base="nn.quantized.modules")
110 self._test_function_import(
120 self._test_function_import(
130 self._test_function_import(
139 self._test_function_import("linear", function_list, base="nn.quantized.modules")
149 self._test_function_import(
[all …]
|
| H A D | test_quantization_fx.py | 24 self._test_function_import("quantize_fx", function_list)
32 self._test_function_import("fx", function_list)
43 self._test_function_import("fx.graph_module", function_list)
54 self._test_function_import("fx.pattern_utils", function_list)
85 self._test_function_import("fx._equalize", function_list)
104 self._test_function_import(
112 self._test_function_import("fx.match_utils", function_list)
116 self._test_function_import("fx.prepare", function_list)
120 self._test_function_import("fx.convert", function_list)
124 self._test_function_import("fx.fuse", function_list)
[all …]
|
| H A D | test_quantization.py | 33 self._test_function_import("quantize", function_list)
41 self._test_function_import("stubs", function_list)
60 self._test_function_import("quantize_jit", function_list)
86 self._test_function_import("fake_quantize", function_list)
99 self._test_function_import("fuse_modules", function_list)
106 self._test_function_import("quant_type", function_list)
137 self._test_function_import("observer", function_list)
162 self._test_function_import("qconfig", function_list)
188 self._test_function_import("quantization_mappings", function_list)
199 self._test_function_import("fuser_method_mappings", function_list)
[all …]
|
| H A D | common.py | 8 def _test_function_import( member in AOMigrationTestCase
|