colin
/
mlir-hlo
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity
30 Commits 1 Branch 0 Tags 2.7 MiB
C++ 59.5%
MLIR 37%
Starlark 2.1%
CMake 0.9%
C 0.2%
Other 0.2%
Go to file
Tres Popp 4251630426 Support CHLO broadcasting operations between scalar and unranked tensors. 
This is done through reshaping the unranked tensor into a 1D ranked tensor which will result in a safe broadcast/indexing logic when the other operand is a scalar.

PiperOrigin-RevId: 322553661
 		2020-07-30 22:34:40 +00:00
include/mlir-hlo Add a transform for Gathers to torch_index_select. 2020-07-30 22:34:32 +00:00
lib Support CHLO broadcasting operations between scalar and unranked tensors. 2020-07-30 22:34:40 +00:00
tests Support CHLO broadcasting operations between scalar and unranked tensors. 2020-07-30 22:34:40 +00:00