Searched defs:SparseXentFunctor (Results 1 – 3 of 3) sorted by relevance
61 struct SparseXentFunctor<GPUDevice, T, Index> { struct62 void operator()(OpKernelContext* ctx, typename TTypes<T>::ConstMatrix logits, in operator ()()
127 struct SparseXentFunctor<CPUDevice, T, Index> { struct128 void operator()(OpKernelContext* ctx, typename TTypes<T>::ConstMatrix logits, in operator ()()
154 struct SparseXentFunctor { struct