ccd70d5717
[MLIR][HLO] Add `rank-specialization-to-scf` pass
...
Currently the lowering is only implemented for the unary case. The n-ary case
will follow.
PiperOrigin-RevId: 374162772
2021-05-17 03:56:23 -07:00
76341f3720
[MLIR][HLO] Add mixed test for `rank-specialization-cluster` pass
...
PiperOrigin-RevId: 373762814
2021-05-14 04:40:40 -07:00
420c42a0a1
[MLIR][HLO] Support CHLO unary operations in rank specialization clustering
...
PiperOrigin-RevId: 373397321
2021-05-12 10:20:43 -07:00
596918a6f1
[MLIR][HLO] Allow rank specialization clustering with `chlo.broadcast_select` op
...
PiperOrigin-RevId: 373379990
2021-05-12 08:56:49 -07:00
875803e5e1
[MLIR][HLO] Add more tests for `rank-specialization-cluster` pass
...
PiperOrigin-RevId: 373343750
2021-05-12 04:46:30 -07:00
313d24bc8f
[MLIR][HLO] Add `rank-specialization-cluster` pass
...
Add a pass to cluster unranked C/HLO operations in one
`chlo.rank_specialization_cluster` op. The C/HLO operations are moved to the
body of the operation. Later passes can use this to rank-specialize all these
operations together.
PiperOrigin-RevId: 373336725
2021-05-12 03:46:01 -07:00
A. Unique TensorFlower