jax
49cfe268 - improve concreteness error message for nn.one_hot (#3656)

Commit

5 years ago

improve concreteness error message for nn.one_hot (#3656) * improve nn.one_hot and jax.numpy.arange errors fixes #3654 * deflake * debug

References

Author

mattjj

Parents