Searched defs:XentFunctor (Results 1 – 3 of 3) sorted by relevance
33 struct XentFunctor<GPUDevice, T> { struct34 void operator()(const GPUDevice &d, in operator ()()51 template struct functor::XentFunctor<GPUDevice, Eigen::half>; member in tensorflow::functor52 template struct functor::XentFunctor<GPUDevice, float>; member in tensorflow::functor53 template struct functor::XentFunctor<GPUDevice, double>; member in tensorflow::functor
120 struct XentFunctor<CPUDevice, T> : XentFunctorBase<CPUDevice, T> {}; struct124 struct XentFunctor<SYCLDevice, T> : XentFunctorBase<SYCLDevice, T> {}; struct
29 struct XentFunctor { struct