Searched defs:TopKFunctor (Results 1 – 13 of 13) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A D | topk_op_gpu_uint16.cu.cc | 25 template struct functor::TopKFunctor<GPUDevice, uint16>; member in tensorflow::functor
26 template struct functor::TopKFunctor<GPUDevice, uint32>; member in tensorflow::functor
|
| H A D | topk_op.cc | 113 struct TopKFunctor<CPUDevice, T> { struct
114 static EIGEN_ALWAYS_INLINE Status Compute( in Compute()
|
| H A D | topk_op.h | 30 struct TopKFunctor { struct
|
| H A D | topk_op_gpu_int64.cu.cc | 25 template struct functor::TopKFunctor<GPUDevice, int64>; member in tensorflow::functor
|
| H A D | topk_op_gpu_uint64.cu.cc | 25 template struct functor::TopKFunctor<GPUDevice, uint64>; member in tensorflow::functor
|
| H A D | topk_op_gpu_double.cu.cc | 25 template struct functor::TopKFunctor<GPUDevice, double>; member in tensorflow::functor
|
| H A D | topk_op_gpu_int8.cu.cc | 25 template struct functor::TopKFunctor<GPUDevice, int8>; member in tensorflow::functor
|
| H A D | topk_op_gpu_half.cu.cc | 25 template struct functor::TopKFunctor<GPUDevice, Eigen::half>; member in tensorflow::functor
|
| H A D | topk_op_gpu_uint8.cu.cc | 25 template struct functor::TopKFunctor<GPUDevice, uint8>; member in tensorflow::functor
|
| H A D | topk_op_gpu_int32.cu.cc | 25 template struct functor::TopKFunctor<GPUDevice, int32>; member in tensorflow::functor
|
| H A D | topk_op_gpu_uint32.cu.cc | 25 template struct functor::TopKFunctor<GPUDevice, uint32>; member in tensorflow::functor
|
| H A D | topk_op_gpu_float.cu.cc | 25 template struct functor::TopKFunctor<GPUDevice, float>; member in tensorflow::functor
|
| H A D | topk_op_gpu_int16.cu.cc | 25 template struct functor::TopKFunctor<GPUDevice, int16>; member in tensorflow::functor
|