Add support for more types in gather op. (#2926)
* Add test for i32 gather * Add support for ints to Gather op * Move helper function to anonymous namespace * Add more types * Use static_cast instead of the old one * Style fix * Skip tests on GPU * Add more tests * Skip tests on gpu * Change bool to char
Showing
This diff is collapsed.
This diff is collapsed.
Please
register
or
sign in
to comment