Support PTQ on Keras3 (#1759)

Signed-off-by: zehao-intel <[email protected]>
intel · Apr 30, 2024 · f67e861 · f67e861
1 parent e3c736f
commit f67e861
Show file tree

Hide file tree

Showing 38 changed files with 2,025 additions and 879 deletions.
diff --git a/neural_compressor/tensorflow/algorithms/static_quant/keras.py b/neural_compressor/tensorflow/algorithms/static_quant/keras.py
diff --git a/neural_compressor/tensorflow/keras/layers/__init__.py b/neural_compressor/tensorflow/keras/layers/__init__.py
@@ -19,6 +19,5 @@
 from neural_compressor.tensorflow.keras.layers.dense import QDense
 from neural_compressor.tensorflow.keras.layers.depthwise_conv2d import QDepthwiseConv2D
 from neural_compressor.tensorflow.keras.layers.pool2d import QAvgPool2D, QMaxPool2D
-from neural_compressor.tensorflow.keras.layers.quantizer import DeQuantize, FakeQuant, Quantize
 from neural_compressor.tensorflow.keras.layers.separable_conv2d import QSeparableConv2D
 from neural_compressor.tensorflow.keras.layers.layer_initializer import layer_initializer_dict