Searched defs:InTopKFunctor (Results 1 – 2 of 2) sorted by relevance
43 struct InTopKFunctor { struct54 struct InTopKFunctor<CPUDevice, T, TargetT> { argument
87 struct InTopKFunctor<GPUDevice, T, TargetT> { struct89 using Dims = Eigen::DSizes<Eigen::Index, ndims>;91 void operator()(OpKernelContext* context, in operator ()()