xla
xla copied to clipboard
Implement MHLO->HLO conversion for entry parameter layout tiles.
Implement MHLO->HLO conversion for entry parameter layout tiles.
Does not support the following cases:
- Parameters with nested layouts and tiles (e.g. tuple of tuples).
- Multi result layouts and tiles.
- Tuple (and nested) result layouts and tiles.