Home
last modified time | relevance | path

Searched refs:SparseXentGradGenerator (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dsparse_xent_op_gpu.cu.cc75 template class generator::SparseXentGradGenerator<float, Index>; \
77 template class generator::SparseXentGradGenerator<Eigen::half, Index>;
Dsparse_xent_op.h97 class SparseXentGradGenerator {
99 EIGEN_DEVICE_FUNC EIGEN_ALWAYS_INLINE SparseXentGradGenerator( in SparseXentGradGenerator() function
237 generator::SparseXentGradGenerator<T, Index> sparse_xent_grad_gen( in Compute()