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 |
||
|---|---|---|
| .. | ||
| chlo_ops.cc | ||
| dialect_registration.cc | ||
| hlo_ops.cc | ||
| infer_fusibility_op_interface.cc | ||
| lhlo_ops.cc | ||