jax
9bb3f79b - Avoid unnecessary fori_loop when calculating the block indices.

Commit

1 year ago

Avoid unnecessary fori_loop when calculating the block indices. PiperOrigin-RevId: 621580324

Author

voutcn

Committer

a-googler

Parents