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

1 Commits

Author SHA1 Message Date
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