fix unbound variable error when not using torch or tf backend (#2)

Refactor to use backend specific gradient functions in tests and merges logic into single function
keras-team · Jan 6, 2025 · 4007769 · 4007769
1 parent 4d97345
commit 4007769
Show file tree

Hide file tree

Showing 5 changed files with 257 additions and 877 deletions.
diff --git a/keras/api/_tf_keras/keras/quantizers/__init__.py b/keras/api/_tf_keras/keras/quantizers/__init__.py
@@ -13,13 +13,8 @@
 from keras.src.quantizers.quantizers import compute_float8_amax_history
 from keras.src.quantizers.quantizers import compute_float8_scale
 from keras.src.quantizers.quantizers import fake_quant_with_min_max_args
-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,
 )
-from keras.src.quantizers.quantizers import (
-    fake_quant_with_min_max_vars_per_channel_gradient,
-)
 from keras.src.quantizers.quantizers import quantize_and_dequantize
diff --git a/keras/api/quantizers/__init__.py b/keras/api/quantizers/__init__.py
@@ -13,13 +13,8 @@
 from keras.src.quantizers.quantizers import compute_float8_amax_history
 from keras.src.quantizers.quantizers import compute_float8_scale
 from keras.src.quantizers.quantizers import fake_quant_with_min_max_args
-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,
 )
-from keras.src.quantizers.quantizers import (
-    fake_quant_with_min_max_vars_per_channel_gradient,
-)
 from keras.src.quantizers.quantizers import quantize_and_dequantize
diff --git a/keras/src/quantizers/__init__.py b/keras/src/quantizers/__init__.py
@@ -7,15 +7,10 @@
 from keras.src.quantizers.quantizers import compute_float8_amax_history
 from keras.src.quantizers.quantizers import compute_float8_scale
 from keras.src.quantizers.quantizers import fake_quant_with_min_max_args
-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,
 )
-from keras.src.quantizers.quantizers import (
-    fake_quant_with_min_max_vars_per_channel_gradient,
-)
 from keras.src.quantizers.quantizers import quantize_and_dequantize
 from keras.src.saving import serialization_lib
 from keras.src.utils.naming import to_snake_case