jax
b1f4ba9d - Prepare TPU interpret mode for AbstractRefs having a memory_space attr.

Commit

219 days ago

Prepare TPU interpret mode for AbstractRefs having a memory_space attr. PiperOrigin-RevId: 776231499

References

Author

jburnim

Committer

Google-ML-Automation

Parents