jax
63d2f7d3 - Declare `tpu.vector_load`, mirroring `tpu.vector_store`.

Commit

286 days ago

Declare `tpu.vector_load`, mirroring `tpu.vector_store`. At the moment this op does not have any lowerings defined and should not be used. PiperOrigin-RevId: 755914724

References

#28289 - Declare `tpu.vector_load`, mirroring `tpu.vector_store`.

Author

nvgrw

Committer

Google-ML-Automation

Parents

59f07c3b

jax 63d2f7d3 - Declare `tpu.vector_load`, mirroring `tpu.vector_store`.

jax
63d2f7d3 - Declare `tpu.vector_load`, mirroring `tpu.vector_store`.