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

  /external/tensorflow/tensorflow/core/kernels/
xent_op_gpu.cu.cc 33 struct XentFunctor<GPUDevice, T> {
46 template struct functor::XentFunctor<GPUDevice, Eigen::half>;
47 template struct functor::XentFunctor<GPUDevice, float>;
48 template struct functor::XentFunctor<GPUDevice, double>;
xent_op.cc 71 functor::XentFunctor<Device, T> functor;
95 struct XentFunctor<CPUDevice, T> : XentFunctorBase<CPUDevice, T> {};
99 struct XentFunctor<SYCLDevice, T> : XentFunctorBase<SYCLDevice, T> {};
xent_op.h 28 struct XentFunctor {
43 // Eigen code implementing XentFunctor::operator().

Completed in 64 milliseconds