supports leakyrelu when negative_slop != 0.

PiperOrigin-RevId: 317395286 Change-Id: I13c433a5d0d2b9ad94792fa28d047a430438158f
google · Jun 19, 2020 · a1fb541 · a1fb541
1 parent 01d9743
commit a1fb541
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/qkeras/qtools/quantized_operators/quantizer_impl.py b/qkeras/qtools/quantized_operators/quantizer_impl.py
@@ -203,6 +203,8 @@ def convert_qkeras_quantizer(
     self.mode = mode
     self.bits = bits
     self.int_bits = int_bits
+    if hasattr(quantizer, "negative_slope") and quantizer.negative_slope != 0:
+      self.is_signed = 1
 
 
 class Ternary(IQuantizer):