jax
c12929b0 - Add more API set up for Mock GPU client. Also clean up previous mock GPU client

Commit

2 years ago

Add more API set up for Mock GPU client. Also clean up previous mock GPU client API. PiperOrigin-RevId: 570153877

Author

wang12tao

Committer

a-googler

Parents