Commit message (Expand) | Author | Age | |
---|---|---|---|
* | Fix cub include path so that TensorFlow compiles when used as a bazel depende... | 2018-09-20 | |
* | Fix error in topk heap launch code. It assumed sizeof(struct) == sizeof(struct | 2017-10-23 | |
* | Update top_k implementation to use iterators for segment offsets | 2017-08-17 | |
* | More WhereOp/TopK GPU bugfixes: use the direct cuda stream for CUB GPU kernel. | 2017-07-07 | |
* | Get TopK op working on GPU again. Extend using cub's radix sort. | 2017-07-06 | |
* | Automated g4 rollback of changelist 156556243 | 2017-05-25 | |
* | Add a GPU implementation for the TopK op. | 2017-05-19 |