| commit | 88fc1d39ff7e956d1fb1eb14812cbaa8051f9fcc | [log] [tgz] |
|---|---|---|
| author | Trevor Killeen <killeent@users.noreply.github.com> | Tue Apr 25 10:39:20 2017 -0400 |
| committer | Soumith Chintala <soumith@gmail.com> | Tue Apr 25 16:39:20 2017 +0200 |
| tree | 9a2e35cfc25812738dd0f6d1687c4987435dcb04 | |
| parent | 8236d38e81396ac48697ac289c0476cff18a8e08 [diff] |
Generic TopK implementation (#744) * move TopK to generic * partial genericization of kernel code * introduce TopKTypeConfig, specialize radix type and conversion for floats * implement topk for byte tensor * implement for char tensor * implement for int tensor, extend test to check indices as well * works for longs too * make bitfield set/get a struct, add support for 64-bit types * extend to double tensor * implement for half tensor * asserts; test fix