colin
/
mlir-hlo
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity
28 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
Robert Suderman c23ad602c8 Add a transform for Gathers to torch_index_select. 
Some gathers can be interpreted as torch index selects. Transforming these
cases allow torch_index_select lowerings to be used for certain gathers.

PiperOrigin-RevId: 322255835
 		2020-07-30 22:34:32 +00:00
include/mlir-hlo Add a transform for Gathers to torch_index_select. 2020-07-30 22:34:32 +00:00
lib Add a transform for Gathers to torch_index_select. 2020-07-30 22:34:32 +00:00
tests Add a transform for Gathers to torch_index_select. 2020-07-30 22:34:32 +00:00