aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/core/kernels/topk_op_gpu.cu.cc
Commit message (Expand)AuthorAge
* Fix cub include path so that TensorFlow compiles when used as a bazel depende...Gravatar A. Unique TensorFlower2018-09-20
* Fix error in topk heap launch code. It assumed sizeof(struct) == sizeof(structGravatar A. Unique TensorFlower2017-10-23
* Update top_k implementation to use iterators for segment offsetsGravatar A. Unique TensorFlower2017-08-17
* More WhereOp/TopK GPU bugfixes: use the direct cuda stream for CUB GPU kernel.Gravatar Eugene Brevdo2017-07-07
* Get TopK op working on GPU again. Extend using cub's radix sort.Gravatar Eugene Brevdo2017-07-06
* Automated g4 rollback of changelist 156556243Gravatar A. Unique TensorFlower2017-05-25
* Add a GPU implementation for the TopK op.Gravatar A. Unique TensorFlower2017-05-19