fix fused_qbias rnd bw corner case

calad0i · Nov 29, 2023 · 8e95f8b · 8e95f8b
1 parent 2ef7b71
commit 8e95f8b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/HGQ/layers/base.py b/src/HGQ/layers/base.py
@@ -219,7 +219,7 @@ def fused_qbias(self):
         mask = tf.reduce_max(self.act_bw, axis=dims, keepdims=False) > 0
 
         if self.paq.rnd_strategy != 3 and self.can_bias_cover_rnd:
-            bias = tf.pow(2., -tf.floor(fbw + 0.5) - 1) + bias  # type: ignore
+            bias = tf.pow(2., -tf.round(fbw) - 1) + bias  # type: ignore
 
         return tf.where(mask, bias, 0.)