Skip to content

Commit a6d5c7b

Browse files
nutsiepullytensorflower-gardener
authored andcommitted
Add QAT support for LeakyReLU
PiperOrigin-RevId: 367072376
1 parent 9193d70 commit a6d5c7b

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tensorflow_model_optimization/python/core/quantization/keras/default_8bit/default_8bit_quantize_registry.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -81,7 +81,7 @@ class Default8BitQuantizeRegistry(
8181
_QuantizeInfo(layers.Softmax, [], []),
8282
# Enable once verified.
8383
# layers.ELU,
84-
# layers.LeakyReLU,
84+
_QuantizeInfo(layers.LeakyReLU, [], [], True),
8585
# layers.PReLU,
8686
# layers.ThresholdedReLU,
8787

0 commit comments

Comments
 (0)