Searched refs:softmax_col_ind (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/sparse/ |
| D | kernels_gpu.cu.cc | 493 const int softmax_begin, const int softmax_end, const int* softmax_col_ind, in CalculateRowSoftmaxGrad() argument
511 const int softmax_col = ldg(softmax_col_ind + i); in CalculateRowSoftmaxGrad()
530 : ldg(softmax_col_ind + softmax_end - 1) + 1; in CalculateRowSoftmaxGrad()
532 const int softmax_col = ldg(softmax_col_ind + i); in CalculateRowSoftmaxGrad()
559 const int rows, const int* softmax_row_ptr, const int* softmax_col_ind, in CSRSparseMatrixSoftmaxGradKernel2D() argument
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
607 softmax_col_ind, softmax, in CSRSparseMatrixSoftmaxGradKernel3D()
627 auto softmax_col_ind = softmax.col_indices().vec<int32>(); in CSRSparseMatrixSoftmaxGradGPUImpl() local
645 softmax_row_ptr.data(), softmax_col_ind.data(), softmax_values.data(), in CSRSparseMatrixSoftmaxGradGPUImpl()
[all …]
|