HomeSort by relevance Sort by last modified time
    Searched refs:TopKFunctor (Results 1 - 3 of 3) sorted by null

  /external/tensorflow/tensorflow/core/kernels/
topk_op.h 30 struct TopKFunctor {
topk_op.cc 90 Status s = functor::TopKFunctor<Device, T>::Compute(
103 struct TopKFunctor<CPUDevice, T> {
250 Status TopKFunctor<GPUDevice, T>::Compute( \
255 extern template struct functor::TopKFunctor<GPUDevice, T>;
topk_op_gpu.cu.cc 535 struct TopKFunctor<GPUDevice, T> {
566 template struct functor::TopKFunctor<GPUDevice, type>;

Completed in 61 milliseconds