Home
last modified time | relevance | path

Searched refs:CSRSparseMatrixSoftmaxGrad (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/sparse/
H A Dsoftmax_op.cc183 functor::CSRSparseMatrixSoftmaxGrad<Device, T> softmax_grad; in Compute()
210 Status CSRSparseMatrixSoftmaxGrad<GPUDevice, T>::operator()( \
214 extern template struct CSRSparseMatrixSoftmaxGrad<GPUDevice, T>;
H A Dkernels.h224 struct CSRSparseMatrixSoftmaxGrad { struct
H A Dkernels_gpu.cu.cc674 Status CSRSparseMatrixSoftmaxGrad<GPUDevice, T>::operator()( \