mct-nightly 2.2.0.20240901.547__py3-none-any.whl → 2.2.0.20240902.511__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.
Files changed (7) hide show
  1. {mct_nightly-2.2.0.20240901.547.dist-info → mct_nightly-2.2.0.20240902.511.dist-info}/METADATA +1 -1
  2. {mct_nightly-2.2.0.20240901.547.dist-info → mct_nightly-2.2.0.20240902.511.dist-info}/RECORD +7 -7
  3. model_compression_toolkit/__init__.py +1 -1
  4. model_compression_toolkit/core/pytorch/reader/graph_builders.py +1 -1
  5. {mct_nightly-2.2.0.20240901.547.dist-info → mct_nightly-2.2.0.20240902.511.dist-info}/LICENSE.md +0 -0
  6. {mct_nightly-2.2.0.20240901.547.dist-info → mct_nightly-2.2.0.20240902.511.dist-info}/WHEEL +0 -0
  7. {mct_nightly-2.2.0.20240901.547.dist-info → mct_nightly-2.2.0.20240902.511.dist-info}/top_level.txt +0 -0
{mct_nightly-2.2.0.20240901.547.dist-info → mct_nightly-2.2.0.20240902.511.dist-info}/METADATA RENAMED
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.1
2
2
  Name: mct-nightly
3
- Version: 2.2.0.20240901.547
3
+ Version: 2.2.0.20240902.511
4
4
  Summary: A Model Compression Toolkit for neural networks
5
5
  Home-page: UNKNOWN
6
6
  License: UNKNOWN
{mct_nightly-2.2.0.20240901.547.dist-info → mct_nightly-2.2.0.20240902.511.dist-info}/RECORD RENAMED
@@ -1,4 +1,4 @@
1
- model_compression_toolkit/__init__.py,sha256=vx_ujbBiTVjR8fsTAWayLzUFa43ew0NzEIUhh8qP8q4,1573
1
+ model_compression_toolkit/__init__.py,sha256=3gZ7KaDN9YVyZ8h3kQ--DTXrObcp81y40HsqykgSoko,1573
2
2
  model_compression_toolkit/constants.py,sha256=i4wYheBkIdQmsQA-axIpcT3YiSO1USNc-jaNiNE8w6E,3920
3
3
  model_compression_toolkit/defaultdict.py,sha256=LSc-sbZYXENMCw3U9F4GiXuv67IKpdn0Qm7Fr11jy-4,2277
4
4
  model_compression_toolkit/logger.py,sha256=3DByV41XHRR3kLTJNbpaMmikL8icd9e1N-nkQAY9oDk,4567
@@ -267,7 +267,7 @@ model_compression_toolkit/core/pytorch/quantizer/__init__.py,sha256=Rf1RcYmelmdZ
267
267
  model_compression_toolkit/core/pytorch/quantizer/fake_quant_builder.py,sha256=D8_CEuFqKAhbUgKaRw7Jlxo0zlqgPTMu6CIIIM4LfS0,7045
268
268
  model_compression_toolkit/core/pytorch/quantizer/lut_fake_quant.py,sha256=uyeBtNokyDUikk-YkDP_mN_2DX0J5oPm3kSfdSUT2Ck,4420
269
269
  model_compression_toolkit/core/pytorch/reader/__init__.py,sha256=Rf1RcYmelmdZmBV5qOKvKWF575ofc06JFQSq83Jz99A,696
270
- model_compression_toolkit/core/pytorch/reader/graph_builders.py,sha256=ESL8k7RLZogTyG_oTTFDmm4RauZvx2gU-UvnOnEsH6Q,15948
270
+ model_compression_toolkit/core/pytorch/reader/graph_builders.py,sha256=NqWPGZzoYSg8I1n5ApgHr5xlwYTBMuvtOyXWjSTrePQ,15956
271
271
  model_compression_toolkit/core/pytorch/reader/node_holders.py,sha256=7XNc7-l1MZPJGcOESvtAwfIMxrU6kvt3YjF5B7qOqK4,1048
272
272
  model_compression_toolkit/core/pytorch/reader/reader.py,sha256=GEJE0QX8XJFWbYCkbRBtzttZtmmuoACLx8gw9KyAQCE,6015
273
273
  model_compression_toolkit/core/pytorch/statistics_correction/__init__.py,sha256=Rf1RcYmelmdZmBV5qOKvKWF575ofc06JFQSq83Jz99A,696
@@ -527,8 +527,8 @@ model_compression_toolkit/xquant/pytorch/model_analyzer.py,sha256=b93o800yVB3Z-i
527
527
  model_compression_toolkit/xquant/pytorch/pytorch_report_utils.py,sha256=bOc-hFL3gdoSM1Th_S2N_-9JJSlPGpZCTx_QLJHS6lg,3388
528
528
  model_compression_toolkit/xquant/pytorch/similarity_functions.py,sha256=CERxq5K8rqaiE-DlwhZBTUd9x69dtYJlkHOPLB54vm8,2354
529
529
  model_compression_toolkit/xquant/pytorch/tensorboard_utils.py,sha256=mkoEktLFFHtEKzzFRn_jCnxjhJolK12TZ5AQeDHzUO8,9767
530
- mct_nightly-2.2.0.20240901.547.dist-info/LICENSE.md,sha256=aYSSIb-5AFPeITTvXm1UAoe0uYBiMmSS8flvXaaFUks,10174
531
- mct_nightly-2.2.0.20240901.547.dist-info/METADATA,sha256=g1E9kgn0rRtSz8mcB9IXNp02mpMGmvzrDNCBj-_JYGs,19718
532
- mct_nightly-2.2.0.20240901.547.dist-info/WHEEL,sha256=eOLhNAGa2EW3wWl_TU484h7q1UNgy0JXjjoqKoxAAQc,92
533
- mct_nightly-2.2.0.20240901.547.dist-info/top_level.txt,sha256=gsYA8juk0Z-ZmQRKULkb3JLGdOdz8jW_cMRjisn9ga4,26
534
- mct_nightly-2.2.0.20240901.547.dist-info/RECORD,,
530
+ mct_nightly-2.2.0.20240902.511.dist-info/LICENSE.md,sha256=aYSSIb-5AFPeITTvXm1UAoe0uYBiMmSS8flvXaaFUks,10174
531
+ mct_nightly-2.2.0.20240902.511.dist-info/METADATA,sha256=LJ1lir-dRUUMItJBfMXAiht4X2oC3UE8dxN3Plcv8pE,19718
532
+ mct_nightly-2.2.0.20240902.511.dist-info/WHEEL,sha256=eOLhNAGa2EW3wWl_TU484h7q1UNgy0JXjjoqKoxAAQc,92
533
+ mct_nightly-2.2.0.20240902.511.dist-info/top_level.txt,sha256=gsYA8juk0Z-ZmQRKULkb3JLGdOdz8jW_cMRjisn9ga4,26
534
+ mct_nightly-2.2.0.20240902.511.dist-info/RECORD,,
model_compression_toolkit/__init__.py CHANGED
@@ -27,4 +27,4 @@ from model_compression_toolkit import data_generation
27
27
  from model_compression_toolkit import pruning
28
28
  from model_compression_toolkit.trainable_infrastructure.keras.load_model import keras_load_quantized_model
29
29
 
30
- __version__ = "2.2.0.20240901.000547"
30
+ __version__ = "2.2.0.20240902.000511"
model_compression_toolkit/core/pytorch/reader/graph_builders.py CHANGED
@@ -148,7 +148,7 @@ def _extract_input_and_output_shapes(_node: Node) -> Tuple[List, List]:
148
148
  if _node.meta[TYPE] == torch.Tensor:
149
149
  output_shape = [list(_node.meta[TENSOR_META].shape)]
150
150
  elif _node.meta[TYPE] in (list, tuple):
151
- output_shape = [list(m.shape) for m in _node.meta[TENSOR_META]]
151
+ output_shape = [list(m.shape) for m in _node.meta.get(TENSOR_META, [])]
152
152
  elif _node.meta[TYPE] == int:
153
153
  output_shape = [[1]]
154
154
  else: