jax
6aae6f43 - Add tpu_sc.subcore_barrier primitive.

Commit

131 days ago

Add tpu_sc.subcore_barrier primitive. PiperOrigin-RevId: 819728612

References

#32611 - Add tpu_sc.subcore_barrier primitive.

Author

brianwa84

brianwa84

Committer

Google-ML-Automation

Google-ML-Automation

Parents

Loading