jax
294d6f89 - Also update custom tpu rule to set unit correctly based on dtype

Commit
5 years ago
Also update custom tpu rule to set unit correctly based on dtype
Author
Erich Elsen
Parents
Loading