tell jax2tf to ignore clz

jax-ml · Mar 20, 2021 · 97aca25 · 97aca25
1 parent 57d5c6a
commit 97aca25
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/jax/experimental/jax2tf/jax2tf.py b/jax/experimental/jax2tf/jax2tf.py
@@ -797,7 +797,7 @@ def _unexpected_primitive(p: core.Primitive, *args, **kwargs):
 
 # Primitives that are not yet implemented must be explicitly declared here.
 tf_not_yet_impl = [
-  "reduce", "rng_uniform",
+  "reduce", "rng_uniform", "clz",
 
   "igamma_grad_a",
   "random_gamma_grad",