ai-edge-quantizer-nightly 0.4.0.dev20250902__py3-none-any.whl → 0.4.0.dev20250903__py3-none-any.whl
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- ai_edge_quantizer/recipe_test.py +6 -0
- {ai_edge_quantizer_nightly-0.4.0.dev20250902.dist-info → ai_edge_quantizer_nightly-0.4.0.dev20250903.dist-info}/METADATA +1 -1
- {ai_edge_quantizer_nightly-0.4.0.dev20250902.dist-info → ai_edge_quantizer_nightly-0.4.0.dev20250903.dist-info}/RECORD +6 -6
- {ai_edge_quantizer_nightly-0.4.0.dev20250902.dist-info → ai_edge_quantizer_nightly-0.4.0.dev20250903.dist-info}/LICENSE +0 -0
- {ai_edge_quantizer_nightly-0.4.0.dev20250902.dist-info → ai_edge_quantizer_nightly-0.4.0.dev20250903.dist-info}/WHEEL +0 -0
- {ai_edge_quantizer_nightly-0.4.0.dev20250902.dist-info → ai_edge_quantizer_nightly-0.4.0.dev20250903.dist-info}/top_level.txt +0 -0
ai_edge_quantizer/recipe_test.py
CHANGED
@@ -14,6 +14,7 @@
|
|
14
14
|
# ==============================================================================
|
15
15
|
|
16
16
|
import os
|
17
|
+
import unittest # pylint: disable=unused-import, required for OSS.
|
17
18
|
|
18
19
|
from absl.testing import parameterized
|
19
20
|
|
@@ -57,6 +58,7 @@ class RecipeTest(parameterized.TestCase):
|
|
57
58
|
self.assertIsNotNone(quantization_result.quantized_model)
|
58
59
|
return quantization_result
|
59
60
|
|
61
|
+
@unittest.skip('skipping due to b/438971945')
|
60
62
|
def test_quantization_from_dynamic_wi8_afp32_func_succeeds(self):
|
61
63
|
quant_result = self._quantize_with_recipe_func(
|
62
64
|
recipe.dynamic_wi8_afp32, self._test_model_path
|
@@ -66,6 +68,7 @@ class RecipeTest(parameterized.TestCase):
|
|
66
68
|
os.path.getsize(self._test_model_path),
|
67
69
|
)
|
68
70
|
|
71
|
+
@unittest.skip('skipping due to b/438971945')
|
69
72
|
def test_quantization_from_dynamic_wi4_afp32_func_succeeds(self):
|
70
73
|
quant_result = self._quantize_with_recipe_func(
|
71
74
|
recipe.dynamic_wi4_afp32, self._test_model_path
|
@@ -75,6 +78,7 @@ class RecipeTest(parameterized.TestCase):
|
|
75
78
|
os.path.getsize(self._test_model_path),
|
76
79
|
)
|
77
80
|
|
81
|
+
@unittest.skip('skipping due to b/438971945')
|
78
82
|
def test_quantization_from_weight_only_wi8_afp32_func_succeeds(self):
|
79
83
|
quant_result = self._quantize_with_recipe_func(
|
80
84
|
recipe.weight_only_wi8_afp32, self._test_model_path
|
@@ -84,6 +88,7 @@ class RecipeTest(parameterized.TestCase):
|
|
84
88
|
os.path.getsize(self._test_model_path),
|
85
89
|
)
|
86
90
|
|
91
|
+
@unittest.skip('skipping due to b/438971945')
|
87
92
|
def test_quantization_from_weight_only_wi4_afp32_func_succeeds(self):
|
88
93
|
quant_result = self._quantize_with_recipe_func(
|
89
94
|
recipe.weight_only_wi4_afp32, self._test_model_path
|
@@ -135,6 +140,7 @@ class RecipeTest(parameterized.TestCase):
|
|
135
140
|
recipe_func=recipe.static_wi8_ai16,
|
136
141
|
),
|
137
142
|
)
|
143
|
+
@unittest.skip('skipping due to b/438971945')
|
138
144
|
def test_recipe_func_and_json_matches(self, recipe_json_path, recipe_func):
|
139
145
|
# Quantize with recipe from function in recipe module.
|
140
146
|
quant_result_from_func = self._quantize_with_recipe_func(
|
@@ -1,6 +1,6 @@
|
|
1
1
|
Metadata-Version: 2.1
|
2
2
|
Name: ai-edge-quantizer-nightly
|
3
|
-
Version: 0.4.0.
|
3
|
+
Version: 0.4.0.dev20250903
|
4
4
|
Summary: A quantizer for advanced developers to quantize converted AI Edge models.
|
5
5
|
Home-page: https://github.com/google-ai-edge/ai-edge-quantizer
|
6
6
|
Keywords: On-Device ML,AI,Google,TFLite,Quantization,LLMs,GenAI
|
@@ -18,7 +18,7 @@ ai_edge_quantizer/quantizer_test.py,sha256=m6f4ayyaF3yQb9i4V0aFAbmGw0OKZ2Zam1RoT
|
|
18
18
|
ai_edge_quantizer/recipe.py,sha256=MEkfQ2Sg3KAE9LAORHWcbjYNPg06EUbwc1d-VspQA2U,6461
|
19
19
|
ai_edge_quantizer/recipe_manager.py,sha256=6dgbE-IZfEetzXH3p3Qm_9eQutNDOpZnMpiaLTbP-ZQ,14744
|
20
20
|
ai_edge_quantizer/recipe_manager_test.py,sha256=H-B75vwPN5ND-nUa3pOXizeHTv4mufPiC5cL_OlDIYU,34040
|
21
|
-
ai_edge_quantizer/recipe_test.py,sha256=
|
21
|
+
ai_edge_quantizer/recipe_test.py,sha256=QisyaTol8JRZFcGOGyee7QRCvqj5VbF4guKWdIoMUOE,6213
|
22
22
|
ai_edge_quantizer/transformation_instruction_generator.py,sha256=O0U2aZcB8aXQgOV8r9g1rGNzDUiuI5Ta53XnxZbVffE,31576
|
23
23
|
ai_edge_quantizer/transformation_instruction_generator_test.py,sha256=KW5-WoTTo9IqLEVnWxVC8ut8eWLi_91xfKgGqVQ9QDk,54635
|
24
24
|
ai_edge_quantizer/transformation_performer.py,sha256=o4J6OUbI0dLoobVYjkOFw5Po3yH0gZJXrfuTIYais4o,13029
|
@@ -70,8 +70,8 @@ ai_edge_quantizer/utils/tfl_interpreter_utils.py,sha256=EoVjI_hplX_Rml3hfRsGmQOi
|
|
70
70
|
ai_edge_quantizer/utils/tfl_interpreter_utils_test.py,sha256=6fjkM-rycZ95L4yfvlr0TN6RlrhfPzxNUYrZaYO_F0A,12013
|
71
71
|
ai_edge_quantizer/utils/validation_utils.py,sha256=oYw33Sg547AqtGw-choPUJmp9SAKkV46J_ddqSsum2Q,3950
|
72
72
|
ai_edge_quantizer/utils/validation_utils_test.py,sha256=V_qNDikPD4OPB-siOLQCWNVWTAu87h2IgNYt7teFd-o,2934
|
73
|
-
ai_edge_quantizer_nightly-0.4.0.
|
74
|
-
ai_edge_quantizer_nightly-0.4.0.
|
75
|
-
ai_edge_quantizer_nightly-0.4.0.
|
76
|
-
ai_edge_quantizer_nightly-0.4.0.
|
77
|
-
ai_edge_quantizer_nightly-0.4.0.
|
73
|
+
ai_edge_quantizer_nightly-0.4.0.dev20250903.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
|
74
|
+
ai_edge_quantizer_nightly-0.4.0.dev20250903.dist-info/METADATA,sha256=6Cvgfzpz_FcQYw7EE9chQoCA9T7qYOxxJB0S1kQy9jc,1535
|
75
|
+
ai_edge_quantizer_nightly-0.4.0.dev20250903.dist-info/WHEEL,sha256=tZoeGjtWxWRfdplE7E3d45VPlLNQnvbKiYnx7gwAy8A,92
|
76
|
+
ai_edge_quantizer_nightly-0.4.0.dev20250903.dist-info/top_level.txt,sha256=8QTfPnFXNVUhScFLaa-NWZMFWMn72M50DVPubpwWB1g,18
|
77
|
+
ai_edge_quantizer_nightly-0.4.0.dev20250903.dist-info/RECORD,,
|
File without changes
|
File without changes
|