jax
d61c10e6 - Improve the error you get when you fail to construct an offloading policy. Or if the policy returns anything that's not a bool or `Recompute`, `Saveable` or `Offloable(...)`.

Commit

1 year ago

Improve the error you get when you fail to construct an offloading policy. Or if the policy returns anything that's not a bool or `Recompute`, `Saveable` or `Offloable(...)`. PiperOrigin-RevId: 796632013

References

#31139 - Improve the error you get when you fail to construct an offloading policy. Or if the policy returns anything that's not a bool or `Recompute`, `Saveable` or `Offloable(...)`.

Author

yashk2810

Committer

Google-ML-Automation

Parents

1cac6f98

jax d61c10e6 - Improve the error you get when you fail to construct an offloading policy. Or if the policy returns anything that's not a bool or `Recompute`, `Saveable` or `Offloable(...)`.

jax
d61c10e6 - Improve the error you get when you fail to construct an offloading policy. Or if the policy returns anything that's not a bool or `Recompute`, `Saveable` or `Offloable(...)`.