[XLA:GPU] Compute computation layout for module

openxla / xla

A machine learning compiler for GPUs, CPUs, and ML accelerators

Apache License 2.0

2.64k stars 418 forks source link

Closed copybara-service[bot] closed 6 days ago

copybara-service[bot] commented 1 week ago

[XLA:GPU] Compute computation layout for module

This allows us to have parameters with non-default layouts