Searched refs:softmax_row_ptr (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/sparse/ |
| D | kernels_gpu.cu.cc | 559 const int rows, const int* softmax_row_ptr, const int* softmax_col_ind, in CSRSparseMatrixSoftmaxGradKernel2D() argument
568 ldg(softmax_row_ptr + row) /*softmax_begin*/, in CSRSparseMatrixSoftmaxGradKernel2D()
569 ldg(softmax_row_ptr + row + 1) /*softmax_end*/, softmax_col_ind, in CSRSparseMatrixSoftmaxGradKernel2D()
580 const int* softmax_row_ptr, const int* softmax_col_ind, const T* softmax, in CSRSparseMatrixSoftmaxGradKernel3D() argument
604 ldg(softmax_row_ptr + row_offset) /*softmax_begin*/, in CSRSparseMatrixSoftmaxGradKernel3D()
606 ldg(softmax_row_ptr + row_offset + 1) /*softmax_end*/, in CSRSparseMatrixSoftmaxGradKernel3D()
626 auto softmax_row_ptr = softmax.row_pointers().vec<int32>(); in CSRSparseMatrixSoftmaxGradGPUImpl() local
639 DCHECK_EQ(rows + 1, softmax_row_ptr.size()); in CSRSparseMatrixSoftmaxGradGPUImpl()
645 softmax_row_ptr.data(), softmax_col_ind.data(), softmax_values.data(), in CSRSparseMatrixSoftmaxGradGPUImpl()
653 DCHECK_EQ((rows + 1) * batch_size, softmax_row_ptr.size()); in CSRSparseMatrixSoftmaxGradGPUImpl()
[all …]
|