ai-edge-quantizer-nightly 0.0.1.dev20250306__py3-none-any.whl → 0.0.1.dev20250308__py3-none-any.whl

ai_edge_quantizer/utils/test_utils.py

@@ -135,3 +135,33 @@ class BaseOpTestCase(parameterized.TestCase):
       )
       for result in signature_result.output_tensors.values():
         self.assertLess(result, output_tolerance)
+
+  def assert_quantization_accuracy_and_size(
+      self,
+      algorithm_key: _AlgorithmName,
+      model_path: str,
+      op_name: _OpName,
+      op_config: _OpQuantConfig,
+      expected_model_size_reduction: float,
+      weight_tolerance: float = 1e-4,
+      output_tolerance: float = 1e-4,
+  ):
+    """Check if the quantization is successful and the result is valid."""
+    validation_result = self.quantize_and_validate(
+        model_path=model_path,
+        algorithm_key=algorithm_key,
+        op_name=op_name,
+        op_config=op_config,
+    )
+    with self.subTest(name='ModelSizeReduction'):
+      self.assert_model_size_reduction_above_min_pct(
+          validation_result, expected_model_size_reduction
+      )
+    with self.subTest(name='WeightsErrors'):
+      self.assert_weights_errors_below_tolerance(
+          validation_result, weight_tolerance
+      )
+    with self.subTest(name='OutputErrors'):
+      self.assert_output_errors_below_tolerance(
+          validation_result, output_tolerance
+      )

{ai_edge_quantizer_nightly-0.0.1.dev20250306.dist-info → ai_edge_quantizer_nightly-0.0.1.dev20250308.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: ai-edge-quantizer-nightly
-Version: 0.0.1.dev20250306
+Version: 0.0.1.dev20250308
 Summary: A quantizer for advanced developers to quantize converted AI Edge models.
 Home-page: https://github.com/google-ai-edge/ai-edge-quantizer
 Keywords: On-Device ML,AI,Google,TFLite,Quantization,LLMs,GenAI

{ai_edge_quantizer_nightly-0.0.1.dev20250306.dist-info → ai_edge_quantizer_nightly-0.0.1.dev20250308.dist-info}/RECORD

@@ -53,15 +53,15 @@ ai_edge_quantizer/transformations/transformation_utils_test.py,sha256=ks81nNvruO
 ai_edge_quantizer/utils/__init__.py,sha256=lpq1g2ayg3lCPLy79t2VicYcnGKw64FfYIj1V7J-4m8,676
 ai_edge_quantizer/utils/calibration_utils.py,sha256=1Fj9MIO6aLZIRgyd4axvZN4S_O64nB_-Miu1WP664js,2536
 ai_edge_quantizer/utils/calibration_utils_test.py,sha256=Z-AcdTieesWFKyKBb08ZXm4Mgu6cvJ4bg2-MJ7hLD10,2856
-ai_edge_quantizer/utils/test_utils.py,sha256=gmq3QeyVuMAgYuGp_uZ2Ei0TOCDqW2jD2GcfgcCX9FA,5183
+ai_edge_quantizer/utils/test_utils.py,sha256=HwZCIpO9fJRAhuN6t6voXKOYQtcioFtt_tpkAlDsAYk,6205
 ai_edge_quantizer/utils/tfl_flatbuffer_utils.py,sha256=pI4JWq_rhp1tRWwNSFMzbWx-3YEQWQir-bM6S88WKIw,10237
 ai_edge_quantizer/utils/tfl_flatbuffer_utils_test.py,sha256=AbyDxoM62k4ojD8gPdkWo--xe5hlX3t0kobQSA80kuk,7740
 ai_edge_quantizer/utils/tfl_interpreter_utils.py,sha256=ZI16i25bAOpnJUBgRg38EH2CuZ55wxyHFScM12RsOwc,12487
 ai_edge_quantizer/utils/tfl_interpreter_utils_test.py,sha256=Op3JxtOqlrjzmYF18jnnstL1k9xiY9kKJ8S2vklKGkc,11327
 ai_edge_quantizer/utils/validation_utils.py,sha256=oYw33Sg547AqtGw-choPUJmp9SAKkV46J_ddqSsum2Q,3950
 ai_edge_quantizer/utils/validation_utils_test.py,sha256=V_qNDikPD4OPB-siOLQCWNVWTAu87h2IgNYt7teFd-o,2934
-ai_edge_quantizer_nightly-0.0.1.dev20250306.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-ai_edge_quantizer_nightly-0.0.1.dev20250306.dist-info/METADATA,sha256=yPtUua8FBDeyCPVB9HTdKU7TYhCXd4HnSEj6NYMRwDQ,1528
-ai_edge_quantizer_nightly-0.0.1.dev20250306.dist-info/WHEEL,sha256=tZoeGjtWxWRfdplE7E3d45VPlLNQnvbKiYnx7gwAy8A,92
-ai_edge_quantizer_nightly-0.0.1.dev20250306.dist-info/top_level.txt,sha256=8QTfPnFXNVUhScFLaa-NWZMFWMn72M50DVPubpwWB1g,18
-ai_edge_quantizer_nightly-0.0.1.dev20250306.dist-info/RECORD,,
+ai_edge_quantizer_nightly-0.0.1.dev20250308.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+ai_edge_quantizer_nightly-0.0.1.dev20250308.dist-info/METADATA,sha256=Tr4AJb4Hqs7-S2fndP3W3WcvyjoQ338CRUbzYgFVuik,1528
+ai_edge_quantizer_nightly-0.0.1.dev20250308.dist-info/WHEEL,sha256=tZoeGjtWxWRfdplE7E3d45VPlLNQnvbKiYnx7gwAy8A,92
+ai_edge_quantizer_nightly-0.0.1.dev20250308.dist-info/top_level.txt,sha256=8QTfPnFXNVUhScFLaa-NWZMFWMn72M50DVPubpwWB1g,18
+ai_edge_quantizer_nightly-0.0.1.dev20250308.dist-info/RECORD,,