jax
3ed9acba - Merge pull request #22669 from hawkinsp:repeat

Commit

1 year ago

Merge pull request #22669 from hawkinsp:repeat PiperOrigin-RevId: 656075132

References

#22669 - Simplify jaxpr for jnp.repeat in scalar repeat case.

Author

a-googler

a-googler

Parents

Loading