jax
ab57bcb7 - [Mosaic GPU] Add a layout inference rule for `vector.ShapeCastOp`.

Commit

166 days ago

[Mosaic GPU] Add a layout inference rule for `vector.ShapeCastOp`. This is the last rule that was missing to delete legacy layout inference. PiperOrigin-RevId: 796358594

References

#31031 - [Mosaic GPU] Add a layout inference rule for `vector.ShapeCastOp`.

Author

bchetioui

Committer

Google-ML-Automation

Parents

d98d979a

jax ab57bcb7 - [Mosaic GPU] Add a layout inference rule for `vector.ShapeCastOp`.

jax
ab57bcb7 - [Mosaic GPU] Add a layout inference rule for `vector.ShapeCastOp`.