ai-edge-quantizer-nightly 0.0.1.dev20250312__py3-none-any.whl → 0.0.1.dev20250313__py3-none-any.whl
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- ai_edge_quantizer/transformations/quantize_tensor.py +2 -2
- ai_edge_quantizer/transformations/quantize_tensor_test.py +2 -2
- {ai_edge_quantizer_nightly-0.0.1.dev20250312.dist-info → ai_edge_quantizer_nightly-0.0.1.dev20250313.dist-info}/METADATA +1 -1
- {ai_edge_quantizer_nightly-0.0.1.dev20250312.dist-info → ai_edge_quantizer_nightly-0.0.1.dev20250313.dist-info}/RECORD +7 -7
- {ai_edge_quantizer_nightly-0.0.1.dev20250312.dist-info → ai_edge_quantizer_nightly-0.0.1.dev20250313.dist-info}/LICENSE +0 -0
- {ai_edge_quantizer_nightly-0.0.1.dev20250312.dist-info → ai_edge_quantizer_nightly-0.0.1.dev20250313.dist-info}/WHEEL +0 -0
- {ai_edge_quantizer_nightly-0.0.1.dev20250312.dist-info → ai_edge_quantizer_nightly-0.0.1.dev20250313.dist-info}/top_level.txt +0 -0
|
@@ -149,7 +149,7 @@ def _perform_blockwise_quantization(
|
|
|
149
149
|
transformation_input.subgraph,
|
|
150
150
|
transformation_input.buffers,
|
|
151
151
|
)
|
|
152
|
-
blockwise_details.
|
|
152
|
+
blockwise_details.scales = scale_tensor_id
|
|
153
153
|
blockwise_details.blockSize = transformation_input.quant_params.block_size
|
|
154
154
|
# blockwise quantization allows optional zero point.
|
|
155
155
|
if transformation_input.quant_params.zero_point is not None:
|
|
@@ -160,7 +160,7 @@ def _perform_blockwise_quantization(
|
|
|
160
160
|
transformation_input.subgraph,
|
|
161
161
|
transformation_input.buffers,
|
|
162
162
|
)
|
|
163
|
-
blockwise_details.
|
|
163
|
+
blockwise_details.zeroPoints = zero_point_tensor_id
|
|
164
164
|
flatbuffer_quantization.details = blockwise_details
|
|
165
165
|
return flatbuffer_quantization
|
|
166
166
|
|
|
@@ -168,8 +168,8 @@ class QuantizeTensorTest(parameterized.TestCase):
|
|
|
168
168
|
)
|
|
169
169
|
self.assertEqual(quant_param.details.blockSize, 32)
|
|
170
170
|
# Check if the scale and zero point tensors are inserted correctly.
|
|
171
|
-
self.assertEqual(quant_param.details.
|
|
172
|
-
self.assertEqual(quant_param.details.
|
|
171
|
+
self.assertEqual(quant_param.details.scales, 9)
|
|
172
|
+
self.assertEqual(quant_param.details.zeroPoints, 10)
|
|
173
173
|
|
|
174
174
|
def test_int4_constant_packed_correctly(self):
|
|
175
175
|
subgraph = self._model.subgraphs[0]
|
|
@@ -1,6 +1,6 @@
|
|
|
1
1
|
Metadata-Version: 2.1
|
|
2
2
|
Name: ai-edge-quantizer-nightly
|
|
3
|
-
Version: 0.0.1.
|
|
3
|
+
Version: 0.0.1.dev20250313
|
|
4
4
|
Summary: A quantizer for advanced developers to quantize converted AI Edge models.
|
|
5
5
|
Home-page: https://github.com/google-ai-edge/ai-edge-quantizer
|
|
6
6
|
Keywords: On-Device ML,AI,Google,TFLite,Quantization,LLMs,GenAI
|
|
@@ -48,8 +48,8 @@ ai_edge_quantizer/transformations/emulated_subchannel.py,sha256=HVaRxoC8PCAvy3xe
|
|
|
48
48
|
ai_edge_quantizer/transformations/emulated_subchannel_test.py,sha256=gZP6u9NdPXl7s19qB_Un8evou9ZZV6I9Gy0E1rdobHM,7722
|
|
49
49
|
ai_edge_quantizer/transformations/quant_insert.py,sha256=jn6HsJaV-sqBiFPY-Aqbd64t8zgcYVkEkZI375x_FWY,3958
|
|
50
50
|
ai_edge_quantizer/transformations/quant_insert_test.py,sha256=X9ptPDvJCFkR5tejKnD1SlHFGPazQTW-wNNMV9MEAuw,10107
|
|
51
|
-
ai_edge_quantizer/transformations/quantize_tensor.py,sha256=
|
|
52
|
-
ai_edge_quantizer/transformations/quantize_tensor_test.py,sha256=
|
|
51
|
+
ai_edge_quantizer/transformations/quantize_tensor.py,sha256=9YaaWR6osxZoyUM8DUPJr_AjWO0QuNhFc65OFnSGzY4,7866
|
|
52
|
+
ai_edge_quantizer/transformations/quantize_tensor_test.py,sha256=XZOollD1jnpCb78gMZx7yocF7RDBSf9HIf-XdG-y_io,9052
|
|
53
53
|
ai_edge_quantizer/transformations/transformation_utils.py,sha256=R42OIbzwQ7JYJ-Qt46jsqwb6u4MfDGiIPCRZCUGLVCw,4664
|
|
54
54
|
ai_edge_quantizer/transformations/transformation_utils_test.py,sha256=xH64SF3UHDh84vYbt-WvmXNjM-Jg-mefES1ACO1tkqw,6269
|
|
55
55
|
ai_edge_quantizer/utils/__init__.py,sha256=lpq1g2ayg3lCPLy79t2VicYcnGKw64FfYIj1V7J-4m8,676
|
|
@@ -62,8 +62,8 @@ ai_edge_quantizer/utils/tfl_interpreter_utils.py,sha256=x2xA2CFPpe_2trcV8v5xGaBE
|
|
|
62
62
|
ai_edge_quantizer/utils/tfl_interpreter_utils_test.py,sha256=Op3JxtOqlrjzmYF18jnnstL1k9xiY9kKJ8S2vklKGkc,11327
|
|
63
63
|
ai_edge_quantizer/utils/validation_utils.py,sha256=oYw33Sg547AqtGw-choPUJmp9SAKkV46J_ddqSsum2Q,3950
|
|
64
64
|
ai_edge_quantizer/utils/validation_utils_test.py,sha256=V_qNDikPD4OPB-siOLQCWNVWTAu87h2IgNYt7teFd-o,2934
|
|
65
|
-
ai_edge_quantizer_nightly-0.0.1.
|
|
66
|
-
ai_edge_quantizer_nightly-0.0.1.
|
|
67
|
-
ai_edge_quantizer_nightly-0.0.1.
|
|
68
|
-
ai_edge_quantizer_nightly-0.0.1.
|
|
69
|
-
ai_edge_quantizer_nightly-0.0.1.
|
|
65
|
+
ai_edge_quantizer_nightly-0.0.1.dev20250313.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
|
|
66
|
+
ai_edge_quantizer_nightly-0.0.1.dev20250313.dist-info/METADATA,sha256=UY2_FI0iSnb7C0s7HBGywzPiASp1GHphmuI9OKM2Ujc,1528
|
|
67
|
+
ai_edge_quantizer_nightly-0.0.1.dev20250313.dist-info/WHEEL,sha256=tZoeGjtWxWRfdplE7E3d45VPlLNQnvbKiYnx7gwAy8A,92
|
|
68
|
+
ai_edge_quantizer_nightly-0.0.1.dev20250313.dist-info/top_level.txt,sha256=8QTfPnFXNVUhScFLaa-NWZMFWMn72M50DVPubpwWB1g,18
|
|
69
|
+
ai_edge_quantizer_nightly-0.0.1.dev20250313.dist-info/RECORD,,
|
|
File without changes
|
|
File without changes
|