Searched defs:TopKFunctor (Results 1 – 11 of 11) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | topk_op.h | 30 struct TopKFunctor { struct
|
| D | topk_op_gpu_int8.cu.cc | 25 template struct functor::TopKFunctor<GPUDevice, int8>; member in tensorflow::functor
|
| D | topk_op_gpu_float.cu.cc | 25 template struct functor::TopKFunctor<GPUDevice, float>; member in tensorflow::functor
|
| D | topk_op_gpu_double.cu.cc | 25 template struct functor::TopKFunctor<GPUDevice, double>; member in tensorflow::functor
|
| D | topk_op_gpu_int16.cu.cc | 25 template struct functor::TopKFunctor<GPUDevice, int16>; member in tensorflow::functor
|
| D | topk_op_gpu_uint16.cu.cc | 25 template struct functor::TopKFunctor<GPUDevice, uint16>; member in tensorflow::functor
|
| D | topk_op_gpu_half.cu.cc | 25 template struct functor::TopKFunctor<GPUDevice, Eigen::half>; member in tensorflow::functor
|
| D | topk_op_gpu_uint8.cu.cc | 25 template struct functor::TopKFunctor<GPUDevice, uint8>; member in tensorflow::functor
|
| D | topk_op_gpu_int32.cu.cc | 25 template struct functor::TopKFunctor<GPUDevice, int32>; member in tensorflow::functor
|
| D | topk_op_gpu_int64.cu.cc | 25 template struct functor::TopKFunctor<GPUDevice, int64>; member in tensorflow::functor
|
| D | topk_op.cc | 103 struct TopKFunctor<CPUDevice, T> { struct
105 Compute(OpKernelContext* context, bool sorted, int k, in Compute()
|