mct-nightly 0.0.0__py3-none-any.whl → 1.1.0.02122021-003117__py3-none-any.whl

{mct_nightly-0.0.0.dist-info → mct_nightly-1.1.0.2122021.post3117.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: mct-nightly
-Version: 0.0.0
+Version: 1.1.0.02122021-003117
 Summary: A Model Compression Toolkit for neural networks
 Home-page: UNKNOWN
 License: UNKNOWN
@@ -11,6 +11,7 @@ Classifier: Operating System :: OS Independent
 Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
 Requires-Python: >=3.6
 Description-Content-Type: text/markdown
+Requires-Dist: tensorflow (==2.5.*)
 Requires-Dist: networkx (==2.5)
 Requires-Dist: tqdm
 Requires-Dist: Pillow
@@ -18,9 +19,9 @@ Requires-Dist: numpy
 Requires-Dist: opencv-python
 Requires-Dist: scikit-image
 Requires-Dist: scikit-learn
+Requires-Dist: tensorflow-model-optimization
 Requires-Dist: tensorboard
 Requires-Dist: PuLP
-Requires-Dist: matplotlib
 
 # Model Compression Toolkit (MCT)
 ![tests](https://github.com/sony/model_optimization/actions/workflows/run_tests_suite.yml/badge.svg)

{mct_nightly-0.0.0.dist-info → mct_nightly-1.1.0.2122021.post3117.dist-info}/RECORD

@@ -1,33 +1,27 @@
-model_compression_toolkit/__init__.py,sha256=B5RFfeWQNuMdLxrY6TZWHcX2O25oc7KfkKRvVYdGMKQ,1892
+model_compression_toolkit/__init__.py,sha256=3Jc6zSKbZJ2oGnNJMKKw54QWKZdtDMGoYLIGbG8IooU,1895
 model_compression_toolkit/common/__init__.py,sha256=LtklANR21DvGL9oO4ila5u6Ft7M83-dqftICub6Vowk,1454
-model_compression_toolkit/common/base_substitutions.py,sha256=FHVBXTdn_sh6QpkdOpZ4IObcoeF49klPvak3T1XSHzs,1656
+model_compression_toolkit/common/base_substitutions.py,sha256=eAQfGVDrzHw6qRCkdm5ENVJLlod633UWCI2qLQyyoCA,1657
 model_compression_toolkit/common/constants.py,sha256=DWcVXmyONazCWZYp6LT-0Kav0bbMxfB8sDWjMRAxqxY,1472
 model_compression_toolkit/common/data_loader.py,sha256=fqTPODRsWZAhkXExE4iP2kgtHv5kOw1UzBojWXIrq7c,4018
 model_compression_toolkit/common/defaultdict.py,sha256=C9LVHx7D0WniqOsoiENkPNdUj49kiQEabESr6FRT5OA,2132
-model_compression_toolkit/common/framework_implementation.py,sha256=mz3WI7wgXoo_jIs_XM-D1uCl7S-gEPml3UKDjAGri3w,9827
 model_compression_toolkit/common/framework_info.py,sha256=mpPvVS8JYa987MYqmfj32iEsLY-vhqY8UGtgoUYK3_k,7723
 model_compression_toolkit/common/logger.py,sha256=RrDE9VCndUrdbQ_4DL5VP8z10fr5BMCJvxswHIvb3Os,4674
-model_compression_toolkit/common/model_builder_mode.py,sha256=G06Tpf-p7vBpa_k0Z8sNdbwhN_QEX-3lf0ajafSrQKA,1488
-model_compression_toolkit/common/model_collector.py,sha256=c-wy-ln9o9FI7jgpS4pXEB3h81XLXwXKPrY0qjdHQuw,4665
-model_compression_toolkit/common/post_training_quantization.py,sha256=T6T9YmoKgXZMdRuLKVZVVviXb7SaefnQ4HNsk5AiWv4,22575
 model_compression_toolkit/common/similarity_analyzer.py,sha256=fB5oGzRY941XONCxnXRy43n4F1qxqXM8EuF31t__Izg,4047
 model_compression_toolkit/common/statistics_collector.py,sha256=cv-qrVM9JY9JLyxfC5G2EzqPAdbOBMN4Gm3Qf9tOhIY,7973
 model_compression_toolkit/common/user_info.py,sha256=OYUdFBma5TKrDLBXEjgo1Oj0RK0zpvkVmP_5ZJ0X-2c,1535
 model_compression_toolkit/common/bias_correction/__init__.py,sha256=vXN-Q5V_3byIScc1j2ePqHBwR60knX1Vy6-Oh-Ke5sk,698
-model_compression_toolkit/common/bias_correction/apply_bias_correction_to_graph.py,sha256=xYQoZ24IuFrm-4c415BCirYbRaDjKl9OnCW4TDUxO7k,3328
-model_compression_toolkit/common/bias_correction/compute_bias_correction_of_graph.py,sha256=UKpQZzK7U8xc5Tu2ZJIRk2b_rNB8I-lJDDIYWvcfr1A,8372
+model_compression_toolkit/common/bias_correction/apply_bias_correction_to_graph.py,sha256=i_Zky82Ow0_dpeSqp7z_5z2Zm-Yx-tQtY8R1RRIUfDE,2870
+model_compression_toolkit/common/bias_correction/compute_bias_correction_of_graph.py,sha256=-pud80aNesybSqxvYSUiHOvK3CEcTcBvQZtxIGXjaNg,7424
 model_compression_toolkit/common/collectors/__init__.py,sha256=vXN-Q5V_3byIScc1j2ePqHBwR60knX1Vy6-Oh-Ke5sk,698
 model_compression_toolkit/common/collectors/base_collector.py,sha256=s1Kne14LcFUajwftMBrusu0wTPHCUQx4TegPcFSWVss,2427
-model_compression_toolkit/common/collectors/histogram_collector.py,sha256=_MDJPsa98rsuN1p8ix3fhoeCg6k9lEZjBvyqTLQ9Fqw,6860
+model_compression_toolkit/common/collectors/histogram_collector.py,sha256=9Ypg35t-zdWSTryA-XvubHSD_PD8ZvJoQ0IliqFm46c,6843
 model_compression_toolkit/common/collectors/mean_collector.py,sha256=x3n2dmZl2EuXh_LgEQnO7YMlL6G7s0nFltsIhXGx6W0,3835
 model_compression_toolkit/common/collectors/min_max_per_channel_collector.py,sha256=ZzJkogbqDkVj_tiS3gzHfTsEkomroeHBcJrXod0ga4k,5031
-model_compression_toolkit/common/gptq/__init__.py,sha256=vXN-Q5V_3byIScc1j2ePqHBwR60knX1Vy6-Oh-Ke5sk,698
-model_compression_toolkit/common/gptq/gptq_config.py,sha256=XDfJgamLEyJKyq4rmhX5gqhM30a1DiJjKqoy3N6t1pM,2787
 model_compression_toolkit/common/graph/__init__.py,sha256=3x8iejFh5xa6tUnOtu5tU7KFOKdRqADhNac89pDItWo,760
-model_compression_toolkit/common/graph/base_graph.py,sha256=jUDItUWIDquKTmEtVtQ-WC2gK8l7QPZKN_1LHF-tmhs,18408
+model_compression_toolkit/common/graph/base_graph.py,sha256=2Bds0yeUA8HtF1Rg2moBwLRN5cG0p7wBjdIEVT3RJo8,18395
 model_compression_toolkit/common/graph/edge.py,sha256=J1ksD-NHXNA_aMqoEQEaR81URLuAht2KNwu_ARwODHE,3712
-model_compression_toolkit/common/graph/graph_matchers.py,sha256=8m51Ve__sQ-joGxNSyQZJImUE5zGgkp2xuSdhWIb1zY,4754
-model_compression_toolkit/common/graph/graph_searches.py,sha256=HYG5f1SN3cKA3Jeq9iUnubk1EJiPD6k-g6ufFTXaIKc,5093
+model_compression_toolkit/common/graph/graph_matchers.py,sha256=2k5yogBwXcxIAENZTIcpZ-lhCe23ZGqnERcd3RmTWik,4860
+model_compression_toolkit/common/graph/graph_searches.py,sha256=o3xcFMtKJMElFGt0wcEMC0c48Qcv0azsF-e__Ti_TKk,5199
 model_compression_toolkit/common/graph/graph_vis.py,sha256=NqWkLW75ap2n-RX_TmooVtxJ63iR3BgtfRR6f4eUWYs,3704
 model_compression_toolkit/common/graph/node.py,sha256=LUi8l1-mzrI-0_USw1Zt-1j6xkHT4tDSAPx9OXBc5pI,7527
 model_compression_toolkit/common/matchers/__init__.py,sha256=vXN-Q5V_3byIScc1j2ePqHBwR60knX1Vy6-Oh-Ke5sk,698
@@ -38,7 +32,7 @@ model_compression_toolkit/common/matchers/function.py,sha256=Aqe0gLQJUSAn-unHYF5
 model_compression_toolkit/common/matchers/node_matcher.py,sha256=fyKrVaC_S1qU-Fx-SJVW0YxEDv7LMHDUZQyfMgbH-xY,2746
 model_compression_toolkit/common/matchers/walk_matcher.py,sha256=wZuTitcKe4CeGizwb4P1_D5o1wd7ZfaKRA_tTng8FeQ,1112
 model_compression_toolkit/common/mixed_precision/__init__.py,sha256=vXN-Q5V_3byIScc1j2ePqHBwR60knX1Vy6-Oh-Ke5sk,698
-model_compression_toolkit/common/mixed_precision/bit_width_setter.py,sha256=DpgZatq-gEhUHfuN4xw3iagiMrNih1poWGd3uVDVW8g,6431
+model_compression_toolkit/common/mixed_precision/bit_width_setter.py,sha256=b47YTTlz3emzcnFLTYUXyO9nG1AOHe1didsx6KqK8EY,6348
 model_compression_toolkit/common/mixed_precision/distance_weighting.py,sha256=3a42gnctiGI5sLDTQj0WjSaaWp8AImM3nMQHNVpyies,2426
 model_compression_toolkit/common/mixed_precision/kpi.py,sha256=eUYXmR7LJC2SG77XI2yy29UkS_2uimxcS03EzXZJox4,1279
 model_compression_toolkit/common/mixed_precision/mixed_precision_quantization_config.py,sha256=fkTsB_-mbwMWfxXJxpStUBGU8Os3_vLI267oeWKS58A,3423
@@ -49,26 +43,26 @@ model_compression_toolkit/common/mixed_precision/search_methods/linear_programmi
 model_compression_toolkit/common/network_editors/__init__.py,sha256=dJWrbHs2NeGYW7Kzy7m7waIw8Ounjd1tiiL0a4nRAos,1235
 model_compression_toolkit/common/network_editors/actions.py,sha256=czJlYQnC6r38XJcpipvcCy2DQNd01QRwkIh_d3BLwSg,13118
 model_compression_toolkit/common/network_editors/edit_network.py,sha256=KsfILh0y_n5kqkaAJDeuHtJ5Pgf4OW6oWjl13ySV0QQ,1723
-model_compression_toolkit/common/network_editors/node_filters.py,sha256=W10GgyW8JicTgc10VjJtbFTuFVHuoOb8HDwWzxl4qdg,3151
+model_compression_toolkit/common/network_editors/node_filters.py,sha256=T3jMUoajr32BxS28NctEzMm5kBxD9nGSFZCaimAhpaI,3152
 model_compression_toolkit/common/quantization/__init__.py,sha256=vXN-Q5V_3byIScc1j2ePqHBwR60knX1Vy6-Oh-Ke5sk,698
 model_compression_toolkit/common/quantization/node_quantization_config.py,sha256=e81Yoqtp6e8saCDZFd2bYJMRtdxQ7aPCorwhYQSws6g,9359
 model_compression_toolkit/common/quantization/quantization_analyzer.py,sha256=xcZ2BPvUZ5ATNStavKjWOcJnvx8OQ7yBG0O_GgcsvuE,2972
 model_compression_toolkit/common/quantization/quantization_config.py,sha256=N2JZFO2AFlAqv9KbKYtIZlfKi2Dsb5RjNCNR8O76DPs,8503
 model_compression_toolkit/common/quantization/quantization_params_fn_selection.py,sha256=WSarcRfTcG2t62JCXIIucbe86oM_856vCb_1ygn-FeM,5448
-model_compression_toolkit/common/quantization/quantize_graph_weights.py,sha256=VopC_NxKd1rWjXhSTigZdnyldEocfsPAyeX-3WtEzkM,2993
-model_compression_toolkit/common/quantization/quantize_node.py,sha256=MatcsOaUX0ogXzD8WH5v7zUc2ITsHBTpBGw3diYT88w,3593
+model_compression_toolkit/common/quantization/quantize_graph_weights.py,sha256=7f9OUmG1BqPT2BmySNh49e3_NjW-MmG34Yun7mkpTew,2698
+model_compression_toolkit/common/quantization/quantize_node.py,sha256=_g4QVfqwR2MjomqZT6N_8lTEAmYu3RKv1-JdcnJD9Mc,3387
 model_compression_toolkit/common/quantization/set_node_quantization_config.py,sha256=MNn4o26souwd1rkdxWbXF97TA_12knNEuRiVaj4vlCw,7729
 model_compression_toolkit/common/quantization/quantization_params_generation/__init__.py,sha256=8wFyT6t-uX0MQ-2QCa4qGZYmquhWroEdBrsren6wrlI,2045
 model_compression_toolkit/common/quantization/quantization_params_generation/kl_selection.py,sha256=Rk7yKNZDXePffzcOxlqK1er8mXkw0k_Qn3TY-Ve18eE,15158
 model_compression_toolkit/common/quantization/quantization_params_generation/kmeans_params.py,sha256=e8E7Htdc00mqATtHpOJUATadDY-LuHSUfJBh1-FLeKw,2648
-model_compression_toolkit/common/quantization/quantization_params_generation/lp_selection.py,sha256=9TIgdYVi0WPCodpk2lJ1BsEjoVTWQ9fINoXGtOkB0bI,6643
+model_compression_toolkit/common/quantization/quantization_params_generation/lp_selection.py,sha256=8ShckR-tkrWTBiJEFhogHhTsShRF5hTTlHHV7OVJWAA,6642
 model_compression_toolkit/common/quantization/quantization_params_generation/lut_kmeans_params.py,sha256=YUkjhYl4rHDtPIOUfMS7gxKiMFWIcIYhR1_g9Rdk6WQ,3559
 model_compression_toolkit/common/quantization/quantization_params_generation/mae_selection.py,sha256=ipfyH29zpSK78C_ktEOdch5O2-fBDHYOd3b4IU4EFfk,5343
 model_compression_toolkit/common/quantization/quantization_params_generation/mse_selection.py,sha256=R2pBjOIeozczdVi-nqfQvy0Auj0QLP0XsgaCIH6DdMk,5420
 model_compression_toolkit/common/quantization/quantization_params_generation/no_clipping.py,sha256=0TTU_rptXNjhldoH_K1-CE8eWEGq-EoSJPLlRM8dxUQ,6466
 model_compression_toolkit/common/quantization/quantization_params_generation/outlier_filter.py,sha256=SeouwEGp5oQ9vZu6OxHKpw3Hu01mqYmCoS-xneLN4kk,1773
 model_compression_toolkit/common/quantization/quantization_params_generation/qparams_activations_computation.py,sha256=4pHn5oX2vfzfiLv2g7YX925wnSnF0qd5o-q0VcLtbSI,3066
-model_compression_toolkit/common/quantization/quantization_params_generation/qparams_computation.py,sha256=OFrboVTE1D1taaP9a1tgY5P2FCvQkJaB4EvHoPZF4WE,4941
+model_compression_toolkit/common/quantization/quantization_params_generation/qparams_computation.py,sha256=OSiSoIsvVsQ0hZKA6TF_P6NTyXgWeODBRX39ef4taio,4733
 model_compression_toolkit/common/quantization/quantization_params_generation/qparams_search.py,sha256=RP1ylMBISrCAxRVZfPoD6RC5msb-Y6bvrJN9lC8xKGk,8565
 model_compression_toolkit/common/quantization/quantization_params_generation/qparams_weights_computation.py,sha256=G_IqFKpn4yVT6Q-uKbtJaF0N0QUxKvHeKdcNEGKlxwc,4943
 model_compression_toolkit/common/quantization/quantizers/__init__.py,sha256=fpP2DhMfJXbHLoq20ziy48ymPyevl5N7P2U1fzlYRSY,699
@@ -76,37 +70,36 @@ model_compression_toolkit/common/quantization/quantizers/kmeans_quantizer.py,sha
 model_compression_toolkit/common/quantization/quantizers/lut_kmeans_quantizer.py,sha256=IhyoB8dWfwuXo4Ow-8iWWCln8lUZh3gX6cGHIZpf3CQ,2797
 model_compression_toolkit/common/quantization/quantizers/power_of_two_quantizer.py,sha256=NI2UG_cn76svg1j2sGNVWOWuRcqcrVGh2Dge8CMpKB0,2108
 model_compression_toolkit/common/quantization/quantizers/quantizers_helpers.py,sha256=muetNju10st84LnjCTLueSsNZxsXtatxGlV77crb6ms,6014
-model_compression_toolkit/common/substitutions/__init__.py,sha256=vXN-Q5V_3byIScc1j2ePqHBwR60knX1Vy6-Oh-Ke5sk,698
-model_compression_toolkit/common/substitutions/apply_substitutions.py,sha256=hfRgeqS873jMnF9CQJCUVMlElYX1tddGSFo1HfEOMuM,1459
 model_compression_toolkit/common/visualization/__init__.py,sha256=fpP2DhMfJXbHLoq20ziy48ymPyevl5N7P2U1fzlYRSY,699
 model_compression_toolkit/common/visualization/tensorboard_writer.py,sha256=_aqRGqgZ7XcWK5gWl8aYFoMMk79K2Ex3c9aRvrqFAMI,16899
-model_compression_toolkit/keras/__init__.py,sha256=fpP2DhMfJXbHLoq20ziy48ymPyevl5N7P2U1fzlYRSY,699
+model_compression_toolkit/keras/__init__.py,sha256=J44ch9RssgFKwQA1XfWDFfA-UiusQmW4jMBNc69QNgQ,754
 model_compression_toolkit/keras/constants.py,sha256=NbsC3VSuZ4Tl28o7FR5PFIJwG5y3iGhHbsqUM90gaFs,1418
-model_compression_toolkit/keras/default_framework_info.py,sha256=_wZv6luoQOk4eaHGRHXasGVOKqZx5nGVwj3VCi4dPec,5188
-model_compression_toolkit/keras/keras_implementation.py,sha256=FoGhblVhLN7aQtEp6VNJn1qBc56f9avlEYWOFlHAxko,10763
-model_compression_toolkit/keras/quantization_facade.py,sha256=hYzQaVrhFcR0x4uBhDrb11_PN4OssRCqDY9Utgoexmg,13440
+model_compression_toolkit/keras/default_framework_info.py,sha256=-gip0vu3wQXoABILprPlAwwexqFhDSBKfSk-_c-h0WM,5187
+model_compression_toolkit/keras/quantization_facade.py,sha256=cc6Zj_J57blCWDM_M6Yhqj1uhR1SYiInArRYZYkml00,28485
 model_compression_toolkit/keras/tensor_marking.py,sha256=lfVqpVW3Pfb1BS7Q2d0yra3dFfKBsqGzy9CjkCfwnZA,5067
 model_compression_toolkit/keras/back2framework/__init__.py,sha256=fpP2DhMfJXbHLoq20ziy48ymPyevl5N7P2U1fzlYRSY,699
 model_compression_toolkit/keras/back2framework/instance_builder.py,sha256=PO_UaPVEWlKqfw8dexOdEeesQF_SUIACT_kGSLoDDHI,3344
-model_compression_toolkit/keras/back2framework/model_builder.py,sha256=b4qnP2hDrLl6DU5gGHzgej7Ol2Wxp6ZdcCd0-Qos3jg,14531
+model_compression_toolkit/keras/back2framework/model_builder.py,sha256=-NZbDh9iYqFWXrE4z7U59i1YV9cLfldeD78TvgdouqE,15292
+model_compression_toolkit/keras/back2framework/model_collector.py,sha256=i1_uvmeQdzOeIc0eaMCq7knYCTvqeZKJNZF9hcV7L1s,4171
 model_compression_toolkit/keras/gradient_ptq/__init__.py,sha256=fpP2DhMfJXbHLoq20ziy48ymPyevl5N7P2U1fzlYRSY,699
 model_compression_toolkit/keras/gradient_ptq/gptq_loss.py,sha256=GGog0LDTb2_XptBjrlNjn2CJUlh6n7T32GNM1bMi5yo,1841
 model_compression_toolkit/keras/gradient_ptq/graph_info.py,sha256=3538-D-4u1K8nXQcOeDyC-8dcP1tEk7pfv2cbRXwvzE,3188
 model_compression_toolkit/keras/gradient_ptq/graph_update.py,sha256=vdbO2ZyXR4tkMf47CCW67bxuxFp4uTrlVYwK_BHN4Pc,3405
-model_compression_toolkit/keras/gradient_ptq/training_wrapper.py,sha256=38n9GKmig5-Ll8EPotJtVGY9h-6Tt9-onB6aq-67LNg,5991
+model_compression_toolkit/keras/gradient_ptq/training_wrapper.py,sha256=W3wZnEskqR8vv8XQT21QAsiAvQF5IIoytACNxCuK-TE,7942
 model_compression_toolkit/keras/graph_substitutions/__init__.py,sha256=fpP2DhMfJXbHLoq20ziy48ymPyevl5N7P2U1fzlYRSY,699
+model_compression_toolkit/keras/graph_substitutions/substituter.py,sha256=mzuIqICMvXJTfpMo_hHZ9xOLOl-yZVtoyrakYUzaQGY,7776
 model_compression_toolkit/keras/graph_substitutions/substitutions/__init__.py,sha256=fpP2DhMfJXbHLoq20ziy48ymPyevl5N7P2U1fzlYRSY,699
 model_compression_toolkit/keras/graph_substitutions/substitutions/activation_decomposition.py,sha256=S-Y3FBIk_IrigIsuJCENa23ZkRSq9PoTDvGe-_8snyY,3886
 model_compression_toolkit/keras/graph_substitutions/substitutions/batchnorm_folding.py,sha256=U2OjwCZ2sfS_OR4SLVjtx5w79Fg6fte5pLyLU_ec_2Q,5190
-model_compression_toolkit/keras/graph_substitutions/substitutions/input_scaling.py,sha256=KSR73RspupgO_ETr7BQWvvjjhRY86VEGtqDSN9ZeEYw,5184
+model_compression_toolkit/keras/graph_substitutions/substitutions/input_scaling.py,sha256=7pcFm_fxf4d7Hg9NLu37a91OPTjqrSreacw6DyO1cE8,6546
 model_compression_toolkit/keras/graph_substitutions/substitutions/mark_activation.py,sha256=uCZfrIWXw6OyihHV5UlGVy9S6HVnO1HHhZ6V5wmEFbM,3038
-model_compression_toolkit/keras/graph_substitutions/substitutions/relu_bound_correction.py,sha256=ltBqOQKwRn3PNIKiFS62fmXi8aa3HA-WLYypWsIHXLs,5167
+model_compression_toolkit/keras/graph_substitutions/substitutions/relu_bound_correction.py,sha256=jyOOhESbfmePHGhS9MnE2pygxBK24PUAEdRychim__s,5636
 model_compression_toolkit/keras/graph_substitutions/substitutions/remove_relu_upper_bound.py,sha256=8wOjEGvTG2nAR5Zq2TSny_eeDep_8sYbDqyOT_xGtbY,2237
 model_compression_toolkit/keras/graph_substitutions/substitutions/scale_equalization.py,sha256=nRMmqGMF-MJlp3dd6hGSSldGWq6P-wCvGyvSRClf_JI,15460
 model_compression_toolkit/keras/graph_substitutions/substitutions/separableconv_decomposition.py,sha256=rL3dqnRXN7vj8PkBcdP73fwVV__BajBTRLMF8ilViBc,7609
 model_compression_toolkit/keras/graph_substitutions/substitutions/shift_negative_activation.py,sha256=iBCF1K2lHCnVlmjaeE6GfbVRHZuH3jmQcGce9DOoYkQ,23234
 model_compression_toolkit/keras/mixed_precision/__init__.py,sha256=vXN-Q5V_3byIScc1j2ePqHBwR60knX1Vy6-Oh-Ke5sk,698
-model_compression_toolkit/keras/mixed_precision/sensitivity_evaluation.py,sha256=aBoWVKmfqRcCpJKLGKASndZe7aJS1X14PLBkXLLTnAg,11477
+model_compression_toolkit/keras/mixed_precision/sensitivity_evaluation.py,sha256=1SnCo59VmcJVwb9gJD5poJrObytcltznGjf9sWniqZw,11399
 model_compression_toolkit/keras/quantizer/__init__.py,sha256=fpP2DhMfJXbHLoq20ziy48ymPyevl5N7P2U1fzlYRSY,699
 model_compression_toolkit/keras/quantizer/base_quantizer.py,sha256=WoD4UrgXcCjZSXPnOX9slxJLrYDYN7go6nKs-bWjfuo,1735
 model_compression_toolkit/keras/quantizer/fake_quant_builder.py,sha256=jtGyNrTGFt6PFoPG9eQA56eIKUDLsulDAsde7o0mx2A,3342
@@ -125,7 +118,7 @@ model_compression_toolkit/keras/quantizer/mixed_precision/selective_quantizer.py
 model_compression_toolkit/keras/quantizer/mixed_precision/selective_weights_quantize_config.py,sha256=zGiaXcUmbiNRIb2Q9E6BocyCL4cyQjuDpfguC9sFVf8,7521
 model_compression_toolkit/keras/reader/__init__.py,sha256=fpP2DhMfJXbHLoq20ziy48ymPyevl5N7P2U1fzlYRSY,699
 model_compression_toolkit/keras/reader/common.py,sha256=KCsvLFOeFltb9NBqQ2nU0MHYVPDBCYDBKF3oeoaangg,2417
-model_compression_toolkit/keras/reader/connectivity_handler.py,sha256=Pj4nmZl0C2K7iIbaC4PYS5Uvd82flvA9ctbI6DYYB7E,11115
+model_compression_toolkit/keras/reader/connectivity_handler.py,sha256=-AYoapd40V25yt1h9MqDzXUDRVUOYho_h6F-U5zDXoY,11139
 model_compression_toolkit/keras/reader/node_builder.py,sha256=N9KSkY7AaZeq0-Zxi5MhKdCsTW2qmFp1yjqk5aPVPpA,3420
 model_compression_toolkit/keras/reader/reader.py,sha256=0i9GT3rTWDDK5fKbtt6PTqFpM59obk8mK0SDXVbycn4,7939
 model_compression_toolkit/keras/reader/nested_model/__init__.py,sha256=fpP2DhMfJXbHLoq20ziy48ymPyevl5N7P2U1fzlYRSY,699
@@ -134,9 +127,9 @@ model_compression_toolkit/keras/reader/nested_model/nested_model_handler.py,sha2
 model_compression_toolkit/keras/reader/nested_model/nodes_merger.py,sha256=iI63gMLuE4_C-8aKSOLT3xRowViI3Je8aN8r5A37vy0,2167
 model_compression_toolkit/keras/reader/nested_model/outputs_merger.py,sha256=bTzVEtXk-7zrtGN7_NZSAKSuCeBWTXd_sPe7RFYgHUQ,2381
 model_compression_toolkit/keras/visualization/__init__.py,sha256=fpP2DhMfJXbHLoq20ziy48ymPyevl5N7P2U1fzlYRSY,699
-model_compression_toolkit/keras/visualization/nn_visualizer.py,sha256=ya6r3l_O5i6RJCYMean8uoKiv8BXOMK9298nI-yBCe4,4789
-mct_nightly-0.0.0.dist-info/LICENSE,sha256=aYSSIb-5AFPeITTvXm1UAoe0uYBiMmSS8flvXaaFUks,10174
-mct_nightly-0.0.0.dist-info/METADATA,sha256=W_28CTJ-lAtI2LQUNPasjNH9w0SkudUidwmZzmTYbhw,5879
-mct_nightly-0.0.0.dist-info/WHEEL,sha256=ewwEueio1C2XeHTvT17n8dZUJgOvyCWCt0WVNLClP9o,92
-mct_nightly-0.0.0.dist-info/top_level.txt,sha256=gsYA8juk0Z-ZmQRKULkb3JLGdOdz8jW_cMRjisn9ga4,26
-mct_nightly-0.0.0.dist-info/RECORD,,
+model_compression_toolkit/keras/visualization/nn_visualizer.py,sha256=AgrgGRHmV8M_NtFH5JzqmjAfBW2Q6PElHWwAq9WRSRU,4726
+mct_nightly-1.1.0.2122021.post3117.dist-info/LICENSE,sha256=aYSSIb-5AFPeITTvXm1UAoe0uYBiMmSS8flvXaaFUks,10174
+mct_nightly-1.1.0.2122021.post3117.dist-info/METADATA,sha256=D53VNuzBWaVqfm1h-O0Na5x7_kiAsa3x-6bBfX8FVEw,5950
+mct_nightly-1.1.0.2122021.post3117.dist-info/WHEEL,sha256=ewwEueio1C2XeHTvT17n8dZUJgOvyCWCt0WVNLClP9o,92
+mct_nightly-1.1.0.2122021.post3117.dist-info/top_level.txt,sha256=gsYA8juk0Z-ZmQRKULkb3JLGdOdz8jW_cMRjisn9ga4,26
+mct_nightly-1.1.0.2122021.post3117.dist-info/RECORD,,

model_compression_toolkit/__init__.py

@@ -13,8 +13,8 @@
 # limitations under the License.
 # ==============================================================================
 
-
-from model_compression_toolkit.common.gptq.gptq_config import GradientPTQConfig
+from model_compression_toolkit.keras.quantization_facade import keras_post_training_quantization, keras_post_training_quantization_mixed_precision
+from model_compression_toolkit.keras.gradient_ptq.training_wrapper import GradientPTQConfig
 from model_compression_toolkit.common.quantization import quantization_config
 from model_compression_toolkit.common.mixed_precision import mixed_precision_quantization_config
 from model_compression_toolkit.common.quantization.quantization_config import QuantizationConfig, \
@@ -27,8 +27,4 @@ from model_compression_toolkit.common.framework_info import FrameworkInfo
 from model_compression_toolkit.common.defaultdict import DefaultDict
 from model_compression_toolkit.common import network_editors as network_editor
 
-from model_compression_toolkit.keras.quantization_facade import keras_post_training_quantization, \
-    keras_post_training_quantization_mixed_precision
-
-
 __version__ = "1.1.0"

model_compression_toolkit/common/base_substitutions.py

@@ -20,6 +20,7 @@ from typing import Any
 from model_compression_toolkit.common.graph.base_graph import Graph
 from model_compression_toolkit.common.matchers.base_matcher import BaseMatcher
 
+
 class BaseSubstitution(ABC):
     """
     Base class for all substitution classes.

model_compression_toolkit/common/bias_correction/apply_bias_correction_to_graph.py

@@ -14,13 +14,13 @@
 # ==============================================================================
 import copy
 
-from model_compression_toolkit.common.framework_implementation import FrameworkImplementation
 from model_compression_toolkit.common.framework_info import FrameworkInfo
 from model_compression_toolkit.common import Graph, Node
+from model_compression_toolkit.keras.constants import BIAS, USE_BIAS
+
 
 def apply_bias_correction_to_graph(graph_to_apply_bias_correction: Graph,
-                                   fw_info: FrameworkInfo,
-                                   fw_impl: FrameworkImplementation) -> Graph:
+                                   fw_info: FrameworkInfo) -> Graph:
     """
     Get a graph, where each node has a final weights quantization configuration (with a bias
     correction term in it), and apply the bias correction for each node in the graph.
@@ -28,7 +28,6 @@ def apply_bias_correction_to_graph(graph_to_apply_bias_correction: Graph,
     Args:
         graph_to_apply_bias_correction: Graph to apply bias correction to.
         fw_info: Framework information (e.g, operators to quantize their weights).
-        fw_impl: FrameworkImplementation object with a specific framework methods implementation.
 
     Returns:
         Graph with bias correction apply to its' nodes.
@@ -40,28 +39,26 @@ def apply_bias_correction_to_graph(graph_to_apply_bias_correction: Graph,
             # If a kernel was quantized and weights bias correction is enabled in n.quantization_cfg,
             # a bias correction term was calculated during model preparation, and is used now in the node's bias term.
             if n.final_weights_quantization_cfg.weights_bias_correction:
-                _apply_bias_correction_to_node(n, fw_impl)
+                _apply_bias_correction_to_node(n)
     return graph
 
 
-def _apply_bias_correction_to_node(node:Node,
-                                   fw_impl: FrameworkImplementation):
+def _apply_bias_correction_to_node(node:Node):
     """
     Set new bias to node using the bias-correction term that is stored in the
     final weights quantization configuration.
 
     Args:
         node: Node to set its corrected bias after bias-correction.
-        fw_impl: FrameworkImplementation object with a specific framework methods implementation.
 
     """
     correction = node.final_weights_quantization_cfg.bias_corrected
 
-    bias = node.get_weights_by_keys(fw_impl.constants.BIAS)  # get original bias from node's weights
+    bias = node.get_weights_by_keys(BIAS)  # get original bias from node's weights
 
     if bias is not None:  # It the layer has bias, we subtract the correction from original bias
-        node.set_weights_by_keys(fw_impl.constants.BIAS, node.get_weights_by_keys(fw_impl.constants.BIAS) - correction)
+        node.set_weights_by_keys(BIAS, node.get_weights_by_keys(BIAS) - correction)
 
     else:  # It the layer has no bias, we consider it as if it has and its value is 0.
-        node.set_weights_by_keys(fw_impl.constants.BIAS, - correction)
-        node.framework_attr[fw_impl.constants.USE_BIAS] = True  # Mark the use_bias attribute of the node.
+        node.set_weights_by_keys(BIAS, - correction)
+        node.framework_attr[USE_BIAS] = True  # Mark the use_bias attribute of the node.

model_compression_toolkit/common/bias_correction/compute_bias_correction_of_graph.py

@@ -12,22 +12,20 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 # ==============================================================================
-
 import copy
 from typing import Any
 
 import numpy as np
 
-from model_compression_toolkit.common.framework_implementation import FrameworkImplementation
 from model_compression_toolkit.common.framework_info import FrameworkInfo
 from model_compression_toolkit.common import Node, Logger, Graph
 from model_compression_toolkit.common.quantization.quantize_node import get_quantized_kernel_by_weights_qc
 from model_compression_toolkit.common.statistics_collector import BaseStatsContainer
+from model_compression_toolkit.keras.constants import KERNEL
 
 
 def compute_bias_correction_of_graph(graph_co_compute_bias: Graph,
-                                     fw_info: FrameworkInfo,
-                                     fw_impl: FrameworkImplementation) -> Graph:
+                                     fw_info: FrameworkInfo) -> Graph:
     """
     For each node in a graph, and for each candidate weights quantization configuration,
     compute the bias-correction term, and store it in the candidate weights quantization configuration.
@@ -36,7 +34,6 @@ def compute_bias_correction_of_graph(graph_co_compute_bias: Graph,
         graph_co_compute_bias: Graph with nodes to compute the bias correction for
         each node's weights quantization configuration candidates.
         fw_info: Framework info like lists of nodes their kernel should quantized.
-        fw_impl: FrameworkImplementation object with a specific framework methods implementation.
 
     Returns:
         Graph with bias correction for each weights quantization configuration candidate
@@ -46,17 +43,13 @@ def compute_bias_correction_of_graph(graph_co_compute_bias: Graph,
     graph = copy.deepcopy(graph_co_compute_bias)
     for n in graph.nodes:
         if fw_info.in_kernel_ops(n):
-            _compute_bias_correction_per_candidate_qc(n,
-                                                      fw_info,
-                                                      graph.get_in_stats_collector(n),
-                                                      fw_impl=fw_impl)
+            _compute_bias_correction_per_candidate_qc(n, fw_info, graph.get_in_stats_collector(n))
     return graph
 
 
 def _compute_bias_correction_per_candidate_qc(node: Node,
                                               fw_info: FrameworkInfo,
-                                              node_in_stats_collector: BaseStatsContainer,
-                                              fw_impl: FrameworkImplementation):
+                                              node_in_stats_collector: BaseStatsContainer):
     """
     For each candidate weights quantization configuration of a given node,
     compute the bias-correction term, and store it in the candidate weights quantization configuration.
@@ -65,7 +58,6 @@ def _compute_bias_correction_per_candidate_qc(node: Node,
         node: Node to compute the bias correction for its different candidates.
         fw_info: Framework info like lists of nodes their kernel should quantized.
         node_in_stats_collector: Statistics collector of the node for the mean per-channel.
-        fw_impl: FrameworkImplementation object with a specific framework methods implementation.
 
     """
 
@@ -74,8 +66,7 @@ def _compute_bias_correction_per_candidate_qc(node: Node,
             if fw_info.in_kernel_ops(node) and weights_qc.enable_weights_quantization:
                 quantized_kernel, io_channels_axes = get_quantized_kernel_by_weights_qc(fw_info,
                                                                                         node,
-                                                                                        weights_qc,
-                                                                                        fw_impl=fw_impl)
+                                                                                        weights_qc)
 
                 # If a kernel was quantized and weights bias correction is enabled in n.quantization_cfg,
                 # a bias correction term is being calculated and used in the node's bias term.
@@ -84,8 +75,7 @@ def _compute_bias_correction_per_candidate_qc(node: Node,
                                                                              node,
                                                                              node_in_stats_collector,
                                                                              io_channels_axes[1],
-                                                                             quantized_kernel,
-                                                                             fw_impl=fw_impl)
+                                                                             quantized_kernel)
 
                     # Store the correction term to use it later,
                     weights_qc.bias_corrected = bias_correction_term
@@ -135,8 +125,7 @@ def _get_bias_correction_term_of_node(input_channels_axis: int,
                                       n: Node,
                                       node_in_stats_collector: BaseStatsContainer,
                                       output_channels_axis: int,
-                                      quantized_kernel: np.ndarray,
-                                      fw_impl: FrameworkImplementation):
+                                      quantized_kernel: np.ndarray):
     """
     Get the bias correction term for a node, using a quantized kernel (which can be quantized
     using any possible bit width)
@@ -147,8 +136,6 @@ def _get_bias_correction_term_of_node(input_channels_axis: int,
         node_in_stats_collector: Input statistics collector of the node.
         output_channels_axis: Index of output channels of the kernel.
         quantized_kernel: Quantized kernel of the node.
-        fw_impl: FrameworkImplementation object with a specific framework methods implementation.
-
 
     Returns:
         Bias-correction term to subtract from the current node's bias.
@@ -163,7 +150,7 @@ def _get_bias_correction_term_of_node(input_channels_axis: int,
             f'Unknown input channel axis for node named: {n.name},'
             f' please update channel mapping function')
     # Compute the bias correction term.
-    correction = _compute_bias_correction(n.get_weights_by_keys(fw_impl.constants.KERNEL),
+    correction = _compute_bias_correction(n.get_weights_by_keys(KERNEL),
                                           quantized_kernel,
                                           node_in_stats_collector,
                                           output_channels_axis,

model_compression_toolkit/common/collectors/histogram_collector.py

@@ -74,7 +74,7 @@ class HistogramCollector(BaseCollector):
             merged_histogram_min = np.min(bins_stack)
             merged_histogram_max = np.max(bins_stack)
             merged_bin_width = np.min(bins_stack[:, 1] - bins_stack[:, 0])
-            merged_histogram_bins = np.arange(merged_histogram_min, merged_histogram_max+merged_bin_width, merged_bin_width)
+            merged_histogram_bins = np.arange(merged_histogram_min, merged_histogram_max, merged_bin_width)
 
             merged_histogram_counts = None
             for histogram in self.__histogram_per_iteration:  # Iterate all collected histograms and merge them

model_compression_toolkit/common/graph/base_graph.py

@@ -12,7 +12,7 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 # ==============================================================================
-from collections import namedtuple
+
 
 from copy import copy, deepcopy
 from typing import List
@@ -29,11 +29,9 @@ from model_compression_toolkit.common.graph.node import Node
 from model_compression_toolkit.common.statistics_collector import BaseStatsContainer
 from model_compression_toolkit.common.statistics_collector import scale_statistics, shift_statistics
 from model_compression_toolkit.common.user_info import UserInformation
+from model_compression_toolkit.keras.reader.connectivity_handler import OutTensor
 from model_compression_toolkit.common.logger import Logger
 
-OutTensor = namedtuple('OutTensor', 'node node_out_index')
-
-
 class Graph(nx.MultiDiGraph, GraphSearches):
     """
     Base graph representing a model to be optimized.

model_compression_toolkit/common/graph/graph_matchers.py

@@ -19,7 +19,9 @@ from typing import Any, List
 from tensorflow.keras.layers import Layer
 
 from model_compression_toolkit.common.graph.node import Node
-from model_compression_toolkit.common.matchers import node_matcher, walk_matcher, edge_matcher
+from model_compression_toolkit.common.matchers import edge_matcher
+from model_compression_toolkit.common.matchers import node_matcher
+from model_compression_toolkit.common.matchers import walk_matcher
 
 
 class NodeOperationMatcher(node_matcher.BaseNodeMatcher):

model_compression_toolkit/common/graph/graph_searches.py

@@ -17,7 +17,9 @@ from abc import ABC
 from typing import List, Any
 
 from model_compression_toolkit.common.graph.node import Node
-from model_compression_toolkit.common.matchers import node_matcher, base_graph_filter, edge_matcher
+from model_compression_toolkit.common.matchers import base_graph_filter
+from model_compression_toolkit.common.matchers import edge_matcher
+from model_compression_toolkit.common.matchers import node_matcher
 from model_compression_toolkit.common.matchers.walk_matcher import WalkMatcherList
 
 

model_compression_toolkit/common/mixed_precision/bit_width_setter.py

@@ -17,7 +17,6 @@ import copy
 
 from typing import Any, List
 
-from model_compression_toolkit.common.quantization.quantization_config import QuantizationConfig
 from model_compression_toolkit.common import Graph, Node
 from model_compression_toolkit.common.framework_info import FrameworkInfo
 from model_compression_toolkit.common.logger import Logger
@@ -26,7 +25,7 @@ from model_compression_toolkit.common.mixed_precision.mixed_precision_quantizati
 
 
 
-def set_bit_widths(quant_config: QuantizationConfig,
+def set_bit_widths(quant_config: MixedPrecisionQuantizationConfig,
                    graph_to_set_bit_widths: Graph,
                    fw_info: FrameworkInfo = None,
                    bit_widths_config: List[int] = None) -> Graph:

model_compression_toolkit/common/network_editors/node_filters.py

@@ -16,6 +16,7 @@
 from typing import Any
 from model_compression_toolkit.common.matchers.node_matcher import BaseNodeMatcher
 
+
 class NodeTypeFilter(BaseNodeMatcher):
     """
     Class NodeNameFilter to check if a node is of a specific type.

model_compression_toolkit/common/quantization/quantization_params_generation/lp_selection.py

@@ -20,7 +20,7 @@ from model_compression_toolkit.common.constants import MIN_THRESHOLD
 from model_compression_toolkit.common.quantization.quantization_params_generation.qparams_search import qparams_selection_tensor_search, qparams_selection_histogram_search
 from model_compression_toolkit.common.similarity_analyzer import compute_lp_norm
 
-from model_compression_toolkit.common.constants import THRESHOLD
+from model_compression_toolkit.keras.constants import THRESHOLD
 
 
 def lp_selection_tensor(tensor_data: np.ndarray,

model_compression_toolkit/common/quantization/quantization_params_generation/qparams_computation.py

@@ -14,7 +14,6 @@
 # ==============================================================================
 from typing import List
 
-from model_compression_toolkit.common.framework_implementation import FrameworkImplementation
 from model_compression_toolkit.common.framework_info import FrameworkInfo
 from model_compression_toolkit.common import Graph, Node, Logger
 from model_compression_toolkit.common.quantization.quantization_params_generation.qparams_activations_computation \
@@ -22,13 +21,13 @@ from model_compression_toolkit.common.quantization.quantization_params_generatio
     get_activations_qparams
 from model_compression_toolkit.common.quantization.quantization_params_generation.qparams_weights_computation import \
     get_weights_qparams, get_channels_axis
+from model_compression_toolkit.keras.constants import KERNEL
 
 
 def calculate_quantization_params(graph: Graph,
                                   fw_info: FrameworkInfo,
                                   nodes: List[Node] = [],
-                                  specific_nodes: bool = False,
-                                  fw_impl: FrameworkImplementation = None):
+                                  specific_nodes: bool = False):
     """
     For a graph, go over its nodes, compute quantization params (for both weights and activations according
     to the given framework info), and create and attach a NodeQuantizationConfig to each node (containing the
@@ -43,7 +42,6 @@ def calculate_quantization_params(graph: Graph,
         graph: Graph to compute its nodes' thresholds.
         nodes: List of nodes to compute their thresholds instead of computing it for all nodes in the graph.
         specific_nodes: Flag to compute thresholds for only specific nodes.
-        fw_impl: FrameworkImplementation with specific framework implementations.
 
     """
 
@@ -59,7 +57,7 @@ def calculate_quantization_params(graph: Graph,
 
             for candidtae_qc in n.candidates_weights_quantization_cfg:
                 output_channels_axis, _ = get_channels_axis(candidtae_qc, fw_info, n.layer_class)
-                weights_params = get_weights_qparams(n.get_weights_by_keys(fw_impl.constants.KERNEL),
+                weights_params = get_weights_qparams(n.get_weights_by_keys(KERNEL),
                                                      candidtae_qc,
                                                      output_channels_axis)
 

model_compression_toolkit/common/quantization/quantize_graph_weights.py

@@ -16,15 +16,14 @@
 import copy
 
 from model_compression_toolkit import common
-from model_compression_toolkit.common.framework_implementation import FrameworkImplementation
 from model_compression_toolkit.common.framework_info import FrameworkInfo
 from model_compression_toolkit.common.graph.base_graph import Graph
 from model_compression_toolkit.common.quantization.quantize_node import get_quantized_kernel_by_weights_qc
+from model_compression_toolkit.keras.constants import KERNEL
 
 
 def quantize_graph_weights(graph_to_quantize: Graph,
-                           fw_info: FrameworkInfo,
-                           fw_impl: FrameworkImplementation) -> Graph:
+                           fw_info: FrameworkInfo) -> Graph:
     """
     Get a graph representing a model, and quantize its nodes' weights.
     Each node is quantized according to the passed framework info and quantization configuration.
@@ -34,7 +33,6 @@ def quantize_graph_weights(graph_to_quantize: Graph,
     Args:
         graph_to_quantize: Graph to quantize its nodes.
         fw_info: Framework information needed for quantizing the graph's nodes' weights and activations.
-        fw_impl: FrameworkImplementation with specific framework implementations.
 
     """
     graph = copy.deepcopy(graph_to_quantize)
@@ -45,14 +43,13 @@ def quantize_graph_weights(graph_to_quantize: Graph,
         if fw_info.in_kernel_ops(n) and n.final_weights_quantization_cfg.enable_weights_quantization:
             quantized_kernel, io_channels_axes = get_quantized_kernel_by_weights_qc(fw_info,
                                                                                     n,
-                                                                                    n.final_weights_quantization_cfg,
-                                                                                    fw_impl=fw_impl)
+                                                                                    n.final_weights_quantization_cfg)
 
             common.Logger.debug(
                 f'Node name: {n.name} has the following quantization params: '
                 f'{str(n.final_weights_quantization_cfg.weights_quantization_params)}')
 
             # Set the kernel node to be the quantized kernel.
-            n.set_weights_by_keys(fw_impl.constants.KERNEL, quantized_kernel)
+            n.set_weights_by_keys(KERNEL, quantized_kernel)
 
     return graph

model_compression_toolkit/common/quantization/quantize_node.py

@@ -18,18 +18,17 @@ import copy
 
 from model_compression_toolkit import common
 from model_compression_toolkit.common import Logger
-from model_compression_toolkit.common.framework_implementation import FrameworkImplementation
 from model_compression_toolkit.common.graph.node import Node
 from model_compression_toolkit.common.framework_info import FrameworkInfo
 from model_compression_toolkit.common.quantization.node_quantization_config import NodeWeightsQuantizationConfig
 from model_compression_toolkit.common.quantization.quantization_params_generation.qparams_weights_computation import \
     get_channels_axis
+from model_compression_toolkit.keras.constants import KERNEL
 
 
 def get_quantized_kernel_by_weights_qc(fw_info:FrameworkInfo,
                                        n:Node,
-                                       weights_qc: NodeWeightsQuantizationConfig,
-                                       fw_impl: FrameworkImplementation):
+                                       weights_qc: NodeWeightsQuantizationConfig):
     """
     For a node and a weights quantization configuration, compute
     the quantized kernel of the node and return it and the input/output channels indices.
@@ -38,7 +37,6 @@ def get_quantized_kernel_by_weights_qc(fw_info:FrameworkInfo,
         fw_info: A FrameworkInfo object Information needed for quantization about the specific framework (e.g., kernel channels indices, groups of layers by how they should be quantized, etc.).
         n: Node to quantize its kernel.
         weights_qc: Weight quantization configuration to use for the quantization.
-        fw_impl: FrameworkImplementation with specific framework implementations.
 
     Returns:
         A quantized kernel of the node using a weights quantization configuration.
@@ -55,7 +53,7 @@ def get_quantized_kernel_by_weights_qc(fw_info:FrameworkInfo,
                                                                   n.layer_class)
 
     Logger.debug(f'quantizing {n.name} with {weights_qc.weights_n_bits} bits')
-    quantized_kernel = weights_qc.weights_quantization_fn(n.get_weights_by_keys(fw_impl.constants.KERNEL),
+    quantized_kernel = weights_qc.weights_quantization_fn(n.get_weights_by_keys(KERNEL),
                                                           n_bits=weights_qc.weights_n_bits,
                                                           signed=True,
                                                           quantization_params=weights_qc.weights_quantization_params,

model_compression_toolkit/keras/__init__.py

@@ -13,3 +13,5 @@
 # limitations under the License.
 # ==============================================================================
 
+
+from model_compression_toolkit.keras import quantizer