transpose fields of GPUBlockSpec with a transform list

Commit

1 year ago

[Pallas/Mosaic GPU] Replace tiling/transpose fields of GPUBlockSpec with a transform list PiperOrigin-RevId: 679079269

Author

apaszke

Committer

Google-ML-Automation

Parents