Searched refs:grad_softmax_row_ptr (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/sparse/ |
| D | kernels_gpu.cu.cc | 560 const T* softmax, const int* grad_softmax_row_ptr, in CSRSparseMatrixSoftmaxGradKernel2D() argument
570 softmax, ldg(grad_softmax_row_ptr + row) /*grad_softmax_begin*/, in CSRSparseMatrixSoftmaxGradKernel2D()
571 ldg(grad_softmax_row_ptr + row + 1) /*grad_softmax_end*/, in CSRSparseMatrixSoftmaxGradKernel2D()
581 const int* grad_softmax_row_ptr, const int* grad_softmax_col_ind, in CSRSparseMatrixSoftmaxGradKernel3D() argument
609 ldg(grad_softmax_row_ptr + row_offset) /*grad_softmax_begin*/, in CSRSparseMatrixSoftmaxGradKernel3D()
611 ldg(grad_softmax_row_ptr + row_offset + 1) /*grad_softmax_end*/, in CSRSparseMatrixSoftmaxGradKernel3D()
630 auto grad_softmax_row_ptr = grad_softmax.row_pointers().vec<int32>(); in CSRSparseMatrixSoftmaxGradGPUImpl() local
640 DCHECK_EQ(rows + 1, grad_softmax_row_ptr.size()); in CSRSparseMatrixSoftmaxGradGPUImpl()
646 grad_softmax_row_ptr.data(), grad_softmax_col_ind.data(), in CSRSparseMatrixSoftmaxGradGPUImpl()
654 DCHECK_EQ((rows + 1) * batch_size, grad_softmax_row_ptr.size()); in CSRSparseMatrixSoftmaxGradGPUImpl()
[all …]
|