ai-edge-quantizer-nightly 0.3.0.dev20250619__py3-none-any.whl → 0.3.0.dev20250621__py3-none-any.whl

ai_edge_quantizer/utils/test_utils.py

@@ -100,6 +100,7 @@ class BaseOpTestCase(parameterized.TestCase):
       num_validation_samples: int = 4,
       num_calibration_samples: Union[int, None] = None,
       error_metric: str = 'mse',
+      int_min_max: Union[tuple[int, int], None] = None,
   ) -> model_validator.ComparisonResult:
     """Quantizes and validates the given model with the given configurations.
 
@@ -112,6 +113,7 @@ class BaseOpTestCase(parameterized.TestCase):
       num_calibration_samples: The number of samples to use for calibration. If
         None then it will be set to num_validation_samples * 8.
       error_metric: The error error_metric to use for validation.
+      int_min_max: The min and max of the integer input range.
 
     Returns:
       The comparison result of the validation.
@@ -129,13 +131,16 @@ class BaseOpTestCase(parameterized.TestCase):
       calibration_data = tfl_interpreter_utils.create_random_normal_input_data(
           quantizer_instance.float_model,
           num_samples=num_calibration_samples,
+          int_min_max=int_min_max,
       )
       calibration_result = quantizer_instance.calibrate(calibration_data)
       quantization_result = quantizer_instance.quantize(calibration_result)
     else:
       quantization_result = quantizer_instance.quantize()
     test_data = tfl_interpreter_utils.create_random_normal_input_data(
-        quantization_result.quantized_model, num_samples=num_validation_samples
+        quantization_result.quantized_model,
+        num_samples=num_validation_samples,
+        int_min_max=int_min_max,
     )
     return quantizer_instance.validate(test_data, error_metric)
 
@@ -185,6 +190,7 @@ class BaseOpTestCase(parameterized.TestCase):
       expected_model_size_reduction: float,
       weight_tolerance: float = 1e-4,
       output_tolerance: float = 1e-4,
+      int_min_max: Union[tuple[int, int], None] = None,
   ):
     """Check if the quantization is successful and the result is valid."""
     validation_result = self.quantize_and_validate(
@@ -192,6 +198,7 @@ class BaseOpTestCase(parameterized.TestCase):
         algorithm_key=algorithm_key,
         op_name=op_name,
         op_config=op_config,
+        int_min_max=int_min_max,
     )
     with self.subTest(name='ModelSizeReduction'):
       self.assert_model_size_reduction_above_min_pct(
@@ -215,8 +222,9 @@ class BaseOpTestCase(parameterized.TestCase):
       num_validation_samples: int = 4,
       num_calibration_samples: Union[int, None] = None,
       output_tolerance: float = 1e-4,
+      int_min_max: Union[tuple[int, int], None] = None,
   ):
-    """Check if the output errors after quantization are within the tolerance."""
+    """Checks if the output errors after quantization are within the tolerance."""
     validation_result = self.quantize_and_validate(
         model_path=model_path,
         algorithm_key=algorithm_key,
@@ -224,6 +232,7 @@ class BaseOpTestCase(parameterized.TestCase):
         num_calibration_samples=num_calibration_samples,
         op_name=op_name,
         op_config=op_config,
+        int_min_max=int_min_max,
     )
     self.assert_output_errors_below_tolerance(
         validation_result, output_tolerance

ai_edge_quantizer/utils/tfl_interpreter_utils.py

@@ -353,6 +353,7 @@ def create_random_dataset(
     input_details: dict[str, Any],
     num_samples: int,
     random_seed: Union[int, np._typing.ArrayLike],
+    int_min_max: Union[tuple[int, int], None] = None,
 ) -> list[dict[str, Any]]:
   """Creates a random normal dataset for given input details.
 
@@ -360,6 +361,7 @@ def create_random_dataset(
     input_details: A dictionary of input details.
     num_samples: The number of samples to generate.
     random_seed: The random seed to use.
+    int_min_max: The min and max of the integer input range.
 
   Returns:
     A list of dictionaries, each containing a sample of input data (for all
@@ -373,7 +375,13 @@ def create_random_dataset(
       dtype = input_tensor["dtype"]
       shape = input_tensor["shape"]
       if dtype in (np.int32, np.int64):
-        new_data = _create_random_integers(rng, shape, dtype)
+        if int_min_max is None:
+          new_data = _create_random_integers(rng, shape, dtype)
+        else:
+          min_value, max_value = int_min_max
+          new_data = _create_random_integers(
+              rng, shape, dtype, min_value, max_value
+          )
       elif dtype in (np.float32, ml_dtypes.bfloat16):
         new_data = _create_random_normal(rng, shape, dtype)
       elif dtype == np.bool:
@@ -389,18 +397,20 @@ def create_random_normal_input_data(
     tflite_model: Union[str, bytes],
     num_samples: int = 4,
     random_seed: int = 666,
+    int_min_max: Union[tuple[int, int], None] = None,
 ) -> dict[str, list[dict[str, Any]]]:
-  """create random dataset following random distribution for signature runner.
+  """Creates a random normal dataset for a signature runner.
 
   Args:
-    tflite_model: TFLite model path or bytearray
-    num_samples: number of input samples to be generated
-    random_seed: random seed to be used for function
+    tflite_model: TFLite model path or bytearray.
+    num_samples: Number of input samples to be generated.
+    random_seed: Random seed to be used for function.
+    int_min_max: The min and max of the integer input range.
 
   Returns:
-    a list of inputs to the given interpreter, for a single interpreter we may
+    A list of inputs to the given interpreter, for a single interpreter we may
     have multiple signatures so each set of inputs is also represented as
-    list
+    list.
   """
   tfl_interpreter = create_tfl_interpreter(tflite_model)
   signature_defs = tfl_interpreter.get_signature_list()
@@ -410,6 +420,6 @@ def create_random_normal_input_data(
     signature_runner = tfl_interpreter.get_signature_runner(signature_key)
     input_details = signature_runner.get_input_details()
     test_data[signature_key] = create_random_dataset(
-        input_details, num_samples, random_seed
+        input_details, num_samples, random_seed, int_min_max
     )
   return test_data

{ai_edge_quantizer_nightly-0.3.0.dev20250619.dist-info → ai_edge_quantizer_nightly-0.3.0.dev20250621.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: ai-edge-quantizer-nightly
-Version: 0.3.0.dev20250619
+Version: 0.3.0.dev20250621
 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.3.0.dev20250619.dist-info → ai_edge_quantizer_nightly-0.3.0.dev20250621.dist-info}/RECORD

@@ -63,15 +63,15 @@ ai_edge_quantizer/transformations/transformation_utils_test.py,sha256=MWgq29t7rv
 ai_edge_quantizer/utils/__init__.py,sha256=lpq1g2ayg3lCPLy79t2VicYcnGKw64FfYIj1V7J-4m8,676
 ai_edge_quantizer/utils/calibration_utils.py,sha256=e3dG7Nm94Ix0hkTWTWPUhEG6a8QR_cAM3PSwblfJV5g,15106
 ai_edge_quantizer/utils/calibration_utils_test.py,sha256=4BlksXl7b4yptL8xPR67hmJCnjhN9V10a2PunzfHrUE,9372
-ai_edge_quantizer/utils/test_utils.py,sha256=Y2pdMvn1k4gmqDo3noJfzx3fJcDHX_1hcsP6oiIz65Y,8240
+ai_edge_quantizer/utils/test_utils.py,sha256=spqUmSNciOKPQHCBkHE7Zo34eMFq_BfBCAnMT3jAulU,8615
 ai_edge_quantizer/utils/tfl_flatbuffer_utils.py,sha256=pZv8FMWyjBSLN5MGJ2K_dZ6oqkJGbp9RI4CfnlPuPII,10830
 ai_edge_quantizer/utils/tfl_flatbuffer_utils_test.py,sha256=K1SbK8q92qYVtiVj0I0GtugsPTkpIpEKv9zakvFV_Sc,8555
-ai_edge_quantizer/utils/tfl_interpreter_utils.py,sha256=EtOv6cpKM_F0uv2bWuSXylYmTeXT6zUc182pw4sdYSI,13889
+ai_edge_quantizer/utils/tfl_interpreter_utils.py,sha256=vTyy6-4PgfFPL3C8uTq_iPFBwdxCjhrWzUiec4DdFPw,14323
 ai_edge_quantizer/utils/tfl_interpreter_utils_test.py,sha256=6fjkM-rycZ95L4yfvlr0TN6RlrhfPzxNUYrZaYO_F0A,12013
 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.3.0.dev20250619.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-ai_edge_quantizer_nightly-0.3.0.dev20250619.dist-info/METADATA,sha256=_1ICby0AKllEQdwCpKgxbguZWlBDy6vIyrNOJ7LhR7s,1528
-ai_edge_quantizer_nightly-0.3.0.dev20250619.dist-info/WHEEL,sha256=tZoeGjtWxWRfdplE7E3d45VPlLNQnvbKiYnx7gwAy8A,92
-ai_edge_quantizer_nightly-0.3.0.dev20250619.dist-info/top_level.txt,sha256=8QTfPnFXNVUhScFLaa-NWZMFWMn72M50DVPubpwWB1g,18
-ai_edge_quantizer_nightly-0.3.0.dev20250619.dist-info/RECORD,,
+ai_edge_quantizer_nightly-0.3.0.dev20250621.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+ai_edge_quantizer_nightly-0.3.0.dev20250621.dist-info/METADATA,sha256=681mJjVZmED7ft7VdtbDu1c43bz96uHnrnB--dlZhQA,1528
+ai_edge_quantizer_nightly-0.3.0.dev20250621.dist-info/WHEEL,sha256=tZoeGjtWxWRfdplE7E3d45VPlLNQnvbKiYnx7gwAy8A,92
+ai_edge_quantizer_nightly-0.3.0.dev20250621.dist-info/top_level.txt,sha256=8QTfPnFXNVUhScFLaa-NWZMFWMn72M50DVPubpwWB1g,18
+ai_edge_quantizer_nightly-0.3.0.dev20250621.dist-info/RECORD,,