jax
641a1d53 - [Pallas] Add support for run_state to cost estimator.

Commit

1 year ago

[Pallas] Add support for run_state to cost estimator. PiperOrigin-RevId: 703543961

References

Author

justinjfu

Committer

Google-ML-Automation

Parents