Searched refs:SparseXentFunctor (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | sparse_xent_op_gpu.cu.cc | 61 struct SparseXentFunctor<GPUDevice, T, Index> { struct
74 template struct functor::SparseXentFunctor<GPUDevice, float, Index>; \
76 template struct functor::SparseXentFunctor<GPUDevice, Eigen::half, Index>; \
|
| D | sparse_xent_op.cc | 92 functor::SparseXentFunctor<Device, T, Index> functor; in Compute()
103 struct SparseXentFunctor<CPUDevice, T, Index> { struct
|
| D | sparse_xent_op.h | 154 struct SparseXentFunctor { struct
|