Skip to content

Commit

Permalink
No public description
Browse files Browse the repository at this point in the history
PiperOrigin-RevId: 617892426
Change-Id: Ifc812ef3ecbc7a2751a035173df58d530a86d621
  • Loading branch information
danielemoro authored and copybara-github committed Mar 21, 2024
1 parent d389659 commit 6be2f5c
Showing 1 changed file with 0 additions and 1 deletion.
1 change: 0 additions & 1 deletion qkeras/quantizers.py
Original file line number Diff line number Diff line change
Expand Up @@ -1428,7 +1428,6 @@ def __call__(self, x):
v = tf.floor(tf.abs(x) / scale + 0.5)
mask = v < levels / 2
z = tf.sign(x) * tf.where(mask, v, tf.ones_like(v) * levels / 2)
print(idx, self.min_po2_exponent, self.max_po2_exponent, m)
scale = _get_least_squares_scale(alpha="auto_po2", x=x, q=z,
scale_axis=self.scale_axis,
elements_per_scale=self.elements_per_scale,
Expand Down

0 comments on commit 6be2f5c

Please sign in to comment.