colin
/
mlir-hlo
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity
1,062 Commits 1 Branch 0 Tags 2.7 MiB
Commit Graph

4 Commits

Author SHA1 Message Date
Mehdi Amini 8c8e81cb69 Fix pass definition to inherit from the TableGen generated base class (NFC) 
PiperOrigin-RevId: 379860210
 2021-06-16 19:05:11 -07:00
A. Unique TensorFlower 82696f8598 [MLIR][HLO] Annotate `mhlo.clamp` and `mhlo.select` as element-wise broadcasting 
The operations allow for a limited form of broadcasting which allows some
operands to be scalars. As such they are neither strictly `Elementwise`, nor
`Broadcasting`. They do fulfill the requirements for `BroadcastingElementwise`
though.

PiperOrigin-RevId: 379719961
 2021-06-16 07:59:26 -07:00
A. Unique TensorFlower b9e45007d5 [MLIR][HLO] Extend broadcast propagation pass to enable more fusion 
Move element-wise operations into assuming regions. This enables fusion
opportunities within the region.

PiperOrigin-RevId: 378362725
 2021-06-09 03:03:37 -07:00
A. Unique TensorFlower c47869f931 [MLIR][HLO] Rename `move-up-dynamic-broadcasts-for-fusion` to `broadcast-propagation` 
PiperOrigin-RevId: 378102608
 2021-06-08 01:51:10 -07:00