Searched refs:grad_softmax_col_ind (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/sparse/ |
| D | kernels_gpu.cu.cc | 495 const int* grad_softmax_col_ind, const T* grad_softmax, T* gradient) { in CalculateRowSoftmaxGrad() argument
512 const int grad_softmax_col = ldg(grad_softmax_col_ind + j); in CalculateRowSoftmaxGrad()
538 : ldg(grad_softmax_col_ind + j); in CalculateRowSoftmaxGrad()
561 const int* grad_softmax_col_ind, const T* grad_softmax, T* gradient) { in CSRSparseMatrixSoftmaxGradKernel2D() argument
572 grad_softmax_col_ind, grad_softmax, gradient); in CSRSparseMatrixSoftmaxGradKernel2D()
581 const int* grad_softmax_row_ptr, const int* grad_softmax_col_ind, in CSRSparseMatrixSoftmaxGradKernel3D() argument
612 grad_softmax_col_ind, grad_softmax, gradient); in CSRSparseMatrixSoftmaxGradKernel3D()
631 auto grad_softmax_col_ind = grad_softmax.col_indices().vec<int32>(); in CSRSparseMatrixSoftmaxGradGPUImpl() local
646 grad_softmax_row_ptr.data(), grad_softmax_col_ind.data(), in CSRSparseMatrixSoftmaxGradGPUImpl()
678 grad_softmax_row_ptr.data(), grad_softmax_col_ind.data(), in CSRSparseMatrixSoftmaxGradGPUImpl()
|