ai-edge-quantizer-nightly 0.3.0.dev20250604__py3-none-any.whl → 0.3.0.dev20250606__py3-none-any.whl

ai_edge_quantizer/utils/tfl_interpreter_utils.py

@@ -17,6 +17,7 @@
 
 from typing import Any, Optional, Union
 
+import ml_dtypes
 import numpy as np
 
 from ai_edge_quantizer import qtyping
@@ -339,6 +340,15 @@ def _create_random_integers(
   return rng.integers(min_value, max_value, size=shape, dtype=dtype)
 
 
+def _create_random_bool(
+    rng: np.random.Generator,
+    shape: tuple[int, ...],
+    dtype: np.dtype,
+) -> dict[str, Any]:
+  """Creates a random bool dataset sample for given input details."""
+  return rng.choice([True, False], size=shape, replace=False).astype(dtype)
+
+
 def create_random_dataset(
     input_details: dict[str, Any],
     num_samples: int,
@@ -364,8 +374,10 @@ def create_random_dataset(
       shape = input_tensor["shape"]
       if dtype in (np.int32, np.int64):
         new_data = _create_random_integers(rng, shape, dtype)
-      elif dtype == np.float32:
+      elif dtype in (np.float32, ml_dtypes.bfloat16):
         new_data = _create_random_normal(rng, shape, dtype)
+      elif dtype == np.bool:
+        new_data = _create_random_bool(rng, shape, dtype)
       else:
         raise ValueError(f"Unsupported dtype: {input_tensor['dtype']}")
       input_data[arg_name] = new_data

{ai_edge_quantizer_nightly-0.3.0.dev20250604.dist-info → ai_edge_quantizer_nightly-0.3.0.dev20250606.dist-info}/METADATA

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

@@ -66,12 +66,12 @@ ai_edge_quantizer/utils/calibration_utils_test.py,sha256=Z-AcdTieesWFKyKBb08ZXm4
 ai_edge_quantizer/utils/test_utils.py,sha256=fXwQ353P7tSy7W4Hs6YskIbCLLaBYGA724hMMbcqCUk,7129
 ai_edge_quantizer/utils/tfl_flatbuffer_utils.py,sha256=zNlR_SJAkDi-EX63O3pNpFLVqSktysScZKgKk1XT3c8,10616
 ai_edge_quantizer/utils/tfl_flatbuffer_utils_test.py,sha256=K1SbK8q92qYVtiVj0I0GtugsPTkpIpEKv9zakvFV_Sc,8555
-ai_edge_quantizer/utils/tfl_interpreter_utils.py,sha256=WoewyiZpaua80oP0tpgyrw5Ws1v7f4vl88vdzS0UjDE,13490
+ai_edge_quantizer/utils/tfl_interpreter_utils.py,sha256=EtOv6cpKM_F0uv2bWuSXylYmTeXT6zUc182pw4sdYSI,13889
 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.dev20250604.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-ai_edge_quantizer_nightly-0.3.0.dev20250604.dist-info/METADATA,sha256=iG1n9JA6Ei_m2GcCgokIMsjnjbhZIUp8wdCy88U2iQQ,1528
-ai_edge_quantizer_nightly-0.3.0.dev20250604.dist-info/WHEEL,sha256=tZoeGjtWxWRfdplE7E3d45VPlLNQnvbKiYnx7gwAy8A,92
-ai_edge_quantizer_nightly-0.3.0.dev20250604.dist-info/top_level.txt,sha256=8QTfPnFXNVUhScFLaa-NWZMFWMn72M50DVPubpwWB1g,18
-ai_edge_quantizer_nightly-0.3.0.dev20250604.dist-info/RECORD,,
+ai_edge_quantizer_nightly-0.3.0.dev20250606.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+ai_edge_quantizer_nightly-0.3.0.dev20250606.dist-info/METADATA,sha256=s7_-COAXeDIYmaeeTtTaoxI_dGf76elQM3A_NK8RHYM,1528
+ai_edge_quantizer_nightly-0.3.0.dev20250606.dist-info/WHEEL,sha256=tZoeGjtWxWRfdplE7E3d45VPlLNQnvbKiYnx7gwAy8A,92
+ai_edge_quantizer_nightly-0.3.0.dev20250606.dist-info/top_level.txt,sha256=8QTfPnFXNVUhScFLaa-NWZMFWMn72M50DVPubpwWB1g,18
+ai_edge_quantizer_nightly-0.3.0.dev20250606.dist-info/RECORD,,