There is no reason to have a multidimensional iota for codegen. This should be canonicalized to a single dimensional iota followed by a broadcast. Changing iota to on a single dimension and a broadcast substantially simplifies implementing iota operations. PiperOrigin-RevId: 320095470 |
||
|---|---|---|
| include/mlir-hlo | ||
| lib | ||
| tests | ||