jax
bb400756 - [Mosaic GPU] Fix argument checking in case the input shape is not a tuple.

Commit

1 year ago

[Mosaic GPU] Fix argument checking in case the input shape is not a tuple. PiperOrigin-RevId: 634044759

References

Author

Google-ML-Automation

Committer

a-googler

Parents