ai-edge-quantizer-nightly 0.0.1.dev20250219__py3-none-any.whl → 0.0.1.dev20250220__py3-none-any.whl

ai_edge_quantizer/algorithms/uniform_quantize/naive_min_max_quantize.py

@@ -189,8 +189,8 @@ def init_qsvs(
   Args:
     op_info: Aggregated information about the op (e.g., quantization config).
     graph_info: Graph information needed to perform quantization for the op.
-    inputs_to_ignore: Input tensor indices to ignore.
-    outputs_to_ignore: Output tensor indices to ignore.
+    inputs_to_ignore: Operand indices to ignore.
+    outputs_to_ignore: Result indices to ignore.
 
   Returns:
     QSVs.
@@ -199,8 +199,8 @@ def init_qsvs(
 
   inputs_to_ignore = inputs_to_ignore or []
   outputs_to_ignore = outputs_to_ignore or []
-  for i, tensor_idx in enumerate(op_info.op.inputs):
-    if tensor_idx != -1 and i not in inputs_to_ignore:
+  for opr_idx, tensor_idx in enumerate(op_info.op.inputs):
+    if tensor_idx != -1 and opr_idx not in inputs_to_ignore:
       tensor = graph_info.subgraph_tensors[tensor_idx]
       tensor_name = tfl_flatbuffer_utils.get_tensor_name(tensor)
       tensor_data = tfl_flatbuffer_utils.get_tensor_data(
@@ -210,8 +210,8 @@ def init_qsvs(
           tensor_data,
           op_info,
       )
-  for i, tensor_idx in enumerate(op_info.op.outputs):
-    if tensor_idx != -1 and i not in outputs_to_ignore:
+  for res_idx, tensor_idx in enumerate(op_info.op.outputs):
+    if tensor_idx != -1 and res_idx not in outputs_to_ignore:
       tensor = graph_info.subgraph_tensors[tensor_idx]
       tensor_name = tfl_flatbuffer_utils.get_tensor_name(tensor)
       tensor_data = tfl_flatbuffer_utils.get_tensor_data(

ai_edge_quantizer/calibrator.py

@@ -283,9 +283,9 @@ class Calibrator:
         op_info = qtyping.OpInfo(op, op_key, subgraph_op_id, op_quant_config)
         # Ignore the input tensors where any dimension of the shape is 0.
         inputs_to_ignore = [
-            idx
-            for idx in op.inputs
-            if not np.all(graph_info.subgraph_tensors[idx].shape)
+            opr_idx
+            for opr_idx, tensor_idx in enumerate(op.inputs)
+            if not np.all(graph_info.subgraph_tensors[tensor_idx].shape)
         ]
         op_qsvs = qsv_init_func(op_info, graph_info, inputs_to_ignore)
         # Step3: initialize tensor qsvs.

{ai_edge_quantizer_nightly-0.0.1.dev20250219.dist-info → ai_edge_quantizer_nightly-0.0.1.dev20250220.dist-info}/METADATA

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

@@ -2,7 +2,7 @@ ai_edge_quantizer/__init__.py,sha256=4pFSkukSwahYyzwqia0yPRyz8TnFQfGRthVJhYpMWas
 ai_edge_quantizer/algorithm_manager.py,sha256=dOW_ex_KShoFaib7auISgsx5wRtxpaxhWrwA2nWgito,6003
 ai_edge_quantizer/algorithm_manager_api.py,sha256=u903TG0s1uIDhJqfeJne3CFl8A93phZrwgV2-hwdcXU,9247
 ai_edge_quantizer/algorithm_manager_api_test.py,sha256=tL_ozYFTsOPX8qGcti0KTz37nVsCxf0SSG5C45SyT-g,7319
-ai_edge_quantizer/calibrator.py,sha256=IqNMnKpZy24Fl_-94AIsp77KzqypMeyHkkxitMBxl58,11325
+ai_edge_quantizer/calibrator.py,sha256=2J-bX0k09A7vZSRnO3eP49YO2uBMUQh6-sk3JRz9fGQ,11363
 ai_edge_quantizer/calibrator_test.py,sha256=hQk61YUvw1X02CDVAddm2n6Dnyk9GWoDgSpO6nuSJiY,11889
 ai_edge_quantizer/conftest.py,sha256=SxCz-5LlRD_lQm4hQc4c6IGG7DS8d7IyEWY9gnscPN0,794
 ai_edge_quantizer/default_policy.py,sha256=TQ9yY8jtrSpMsTBsTyKW6TY-voGH_psvwGZoFglAbiA,9079
@@ -30,7 +30,7 @@ ai_edge_quantizer/algorithms/nonlinear_quantize/float_casting_test.py,sha256=s64
 ai_edge_quantizer/algorithms/uniform_quantize/__init__.py,sha256=lpq1g2ayg3lCPLy79t2VicYcnGKw64FfYIj1V7J-4m8,676
 ai_edge_quantizer/algorithms/uniform_quantize/common_quantize.py,sha256=wPZevOuowJczG9t4Gynzv7tIeH6zhOnaKPsfr2K_fsk,21259
 ai_edge_quantizer/algorithms/uniform_quantize/common_quantize_test.py,sha256=qMmKbWqxrCoVKbLKHn9WuCrGKPfHkEyU0Nmhokh8Qeo,2597
-ai_edge_quantizer/algorithms/uniform_quantize/naive_min_max_quantize.py,sha256=g69t3oTl1fdphsWLIxVNjuIIIA3lLQ56Vt8Qgo7SocU,9946
+ai_edge_quantizer/algorithms/uniform_quantize/naive_min_max_quantize.py,sha256=WaN6w-DqQkSwNl8xsbsSPPY97oKohHpo-5Ng_5yAerw,9958
 ai_edge_quantizer/algorithms/uniform_quantize/naive_min_max_quantize_test.py,sha256=3UV1I-to_u6NE_yKoXOVUOQgil-tMY6VQ_L273lMfqQ,5949
 ai_edge_quantizer/algorithms/uniform_quantize/uniform_quantize_tensor.py,sha256=B-s1KMfb9tqvaDhHJV-M2zRR078z5Mwv-P9h77S3Mis,12229
 ai_edge_quantizer/algorithms/uniform_quantize/uniform_quantize_tensor_test.py,sha256=WZ4_bvbG999nOtCIqn7mrMnpRdoJOdiyzxhsL_QiPHA,11395
@@ -58,8 +58,8 @@ ai_edge_quantizer/utils/tfl_interpreter_utils.py,sha256=SM8H4i7Jq_nfdsJpImopHndN
 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.dev20250219.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-ai_edge_quantizer_nightly-0.0.1.dev20250219.dist-info/METADATA,sha256=I42-1mg57Fu0eGeUYlGuWKyXvQuQ7l_PzzcOCcc7CAk,1484
-ai_edge_quantizer_nightly-0.0.1.dev20250219.dist-info/WHEEL,sha256=tZoeGjtWxWRfdplE7E3d45VPlLNQnvbKiYnx7gwAy8A,92
-ai_edge_quantizer_nightly-0.0.1.dev20250219.dist-info/top_level.txt,sha256=8QTfPnFXNVUhScFLaa-NWZMFWMn72M50DVPubpwWB1g,18
-ai_edge_quantizer_nightly-0.0.1.dev20250219.dist-info/RECORD,,
+ai_edge_quantizer_nightly-0.0.1.dev20250220.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+ai_edge_quantizer_nightly-0.0.1.dev20250220.dist-info/METADATA,sha256=wc1t3VKLcToSVZ6MOwmrcNhWcy967d9mAaQlFF6w50s,1484
+ai_edge_quantizer_nightly-0.0.1.dev20250220.dist-info/WHEEL,sha256=tZoeGjtWxWRfdplE7E3d45VPlLNQnvbKiYnx7gwAy8A,92
+ai_edge_quantizer_nightly-0.0.1.dev20250220.dist-info/top_level.txt,sha256=8QTfPnFXNVUhScFLaa-NWZMFWMn72M50DVPubpwWB1g,18
+ai_edge_quantizer_nightly-0.0.1.dev20250220.dist-info/RECORD,,