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

3 Commits

Author SHA1 Message Date
Benjamin Kramer d77b556822 [MLIR][MHLO] Allow recursion in the shape_of mover 
This allows it to push shape_of over a chain of ops all the way to the top.

PiperOrigin-RevId: 362249009
 2021-03-11 02:52:21 -08:00
Benjamin Kramer 67a770e4e0 [HLO:MLIR] Make binary op type reification emit shape_of instead of tensor ops 
This gives cleaner code and allows shape optimizations to happen on the result.

PiperOrigin-RevId: 362242975
 2021-03-11 02:01:35 -08:00
A. Unique TensorFlower c217a6ef61 [MHLO] Add pass to move up dynamic broadcasts for fusion 
For now, the pass only reifies the required shape computations. Moving
broadcasts will follow to allow for fusion across them.

PiperOrigin-RevId: 362033715
 2021-03-10 06:21:57 -08:00