fix public api

edgeimpulse · Dec 19, 2024 · e233acf · e233acf
1 parent 4fe6c59
commit e233acf
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 0 deletions.
diff --git a/keras/api/_tf_keras/keras/quantizers/__init__.py b/keras/api/_tf_keras/keras/quantizers/__init__.py
@@ -16,6 +16,7 @@
 from keras.src.quantizers.quantizers import (
     fake_quant_with_min_max_args_gradient,
 )
+from keras.src.quantizers.quantizers import fake_quant_with_min_max_vars
 from keras.src.quantizers.quantizers import (
     fake_quant_with_min_max_vars_per_channel,
 )

diff --git a/keras/api/quantizers/__init__.py b/keras/api/quantizers/__init__.py
@@ -16,6 +16,7 @@
 from keras.src.quantizers.quantizers import (
     fake_quant_with_min_max_args_gradient,
 )
+from keras.src.quantizers.quantizers import fake_quant_with_min_max_vars
 from keras.src.quantizers.quantizers import (
     fake_quant_with_min_max_vars_per_channel,
 )