jax
944d822c - Add a no-op batching rule for optimization_barrier_p

Commit

1 year ago

Add a no-op batching rule for optimization_barrier_p PiperOrigin-RevId: 704507586

References

Author

yashk2810

Committer

Google-ML-Automation

Parents