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

ai_edge_quantizer/transformations/transformation_utils.py

@@ -15,6 +15,7 @@
 
 """Utility functions for graph transformations."""
 
+import copy
 import dataclasses
 from typing import Optional, Union
 
@@ -105,7 +106,17 @@ def get_constant_buffer(
   Returns:
     The index of the new buffer in the model.
   """
-  new_data = np.frombuffer(data.tobytes(), dtype=np.uint8).flatten()
+
+  if isinstance(data, np.ndarray):
+    # in the case where the data is passed from quantization_params.
+    new_data = np.frombuffer(data.tobytes(), dtype=np.uint8).flatten()
+  elif isinstance(data, bytes):
+    # in the case where the data is coming from duplicating buffers, we need to
+    # make a copy of the data to avoid having two buffers pointing to the same
+    # data.
+    new_data = copy.deepcopy(data)
+  else:
+    raise ValueError('data passed in must be either np.ndarray or bytes.')
   # TODO: b/417811116 - we should make this more efficient.
   if not force_duplicate_buffer:
     for index, buffer in enumerate(buffers):

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

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

@@ -58,7 +58,7 @@ ai_edge_quantizer/transformations/quant_insert.py,sha256=jn6HsJaV-sqBiFPY-Aqbd64
 ai_edge_quantizer/transformations/quant_insert_test.py,sha256=X9ptPDvJCFkR5tejKnD1SlHFGPazQTW-wNNMV9MEAuw,10107
 ai_edge_quantizer/transformations/quantize_tensor.py,sha256=kjaNrw9mnrn0t8u0vey9S_uPz3iVUicwy4rluxVqV3E,7617
 ai_edge_quantizer/transformations/quantize_tensor_test.py,sha256=mHLO3_MRt36A8-ZN8ADn5tBBJlqjTWa7ZUN8Mmu5Rcw,9116
-ai_edge_quantizer/transformations/transformation_utils.py,sha256=mYD6gHxSO6JmjC9fSWl0pLh6N_tCW70yTIxHYdmnV3A,6483
+ai_edge_quantizer/transformations/transformation_utils.py,sha256=GwIaKVsePZYgVG2lSanOswcaZYMjvgyqstDVwXl9DGY,6923
 ai_edge_quantizer/transformations/transformation_utils_test.py,sha256=MWgq29t7rvxRQIfi4ny9IoODFCTcbpjnIwoCL40zDKk,8698
 ai_edge_quantizer/utils/__init__.py,sha256=lpq1g2ayg3lCPLy79t2VicYcnGKw64FfYIj1V7J-4m8,676
 ai_edge_quantizer/utils/calibration_utils.py,sha256=1Fj9MIO6aLZIRgyd4axvZN4S_O64nB_-Miu1WP664js,2536
@@ -70,8 +70,8 @@ ai_edge_quantizer/utils/tfl_interpreter_utils.py,sha256=WoewyiZpaua80oP0tpgyrw5W
 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.dev20250603.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-ai_edge_quantizer_nightly-0.3.0.dev20250603.dist-info/METADATA,sha256=h-79xv1FAsiMP50ipV6hhK8T0asiofo5-LJt-TlROpc,1528
-ai_edge_quantizer_nightly-0.3.0.dev20250603.dist-info/WHEEL,sha256=tZoeGjtWxWRfdplE7E3d45VPlLNQnvbKiYnx7gwAy8A,92
-ai_edge_quantizer_nightly-0.3.0.dev20250603.dist-info/top_level.txt,sha256=8QTfPnFXNVUhScFLaa-NWZMFWMn72M50DVPubpwWB1g,18
-ai_edge_quantizer_nightly-0.3.0.dev20250603.dist-info/RECORD,,
+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,,