INTERNAL CHANGE.

PiperOrigin-RevId: 434514577
Raverss · Mar 14, 2022 · 97d254b · 97d254b
1 parent a632846
commit 97d254b
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/aqt/jax/flax_layers.py b/aqt/jax/flax_layers.py
@@ -170,6 +170,7 @@ def __call__(
     inputs = jnp.asarray(inputs, self.dtype)
     kernel = jnp.asarray(kernel, self.dtype)
 
+
     get_bounds_params = get_bounds.GetBounds.Params(
         update_bounds=self.quant_context.update_bounds,
         update_stats=self.train,