Commits


RandySheriffH authored and GitHub committed d6849bd26cf
Rashuai/cuda top k (#1919) * implement cuda topk * implement heap * add type support * refactor interface * add support for sorting by index * add test case * use cub device radix sort * register for opset 9 and 10 * add opset 9/10 delaration * refactor code * refactor code * fix comment * fix comment * switch to scratched mem