Searched refs:TopKFunctor (Results 1 – 12 of 12) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | topk_op.cc | 90 Status s = functor::TopKFunctor<Device, T>::Compute( in Compute()
103 struct TopKFunctor<CPUDevice, T> { struct
250 Status TopKFunctor<GPUDevice, T>::Compute( \
255 extern template struct functor::TopKFunctor<GPUDevice, T>;
|
| 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_int8.cu.cc | 25 template struct functor::TopKFunctor<GPUDevice, int8>; 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_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_gpu_half.cu.cc | 25 template struct functor::TopKFunctor<GPUDevice, Eigen::half>; member in tensorflow::functor
|
| D | topk_op.h | 30 struct TopKFunctor { struct
|
| D | topk_op_gpu.h | 538 struct TopKFunctor<GPUDevice, T> {
|