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