Searched defs:grad_softmax (Results 1 – 4 of 4) sorted by relevance
78 def _SparseMatrixSoftmaxGrad(op, grad_softmax): argument
482 const int* grad_softmax_col_ind, const T* grad_softmax, T* gradient) { in CalculateRowSoftmaxGrad()548 const int* grad_softmax_col_ind, const T* grad_softmax, T* gradient) { in CSRSparseMatrixSoftmaxGradKernel2D()569 const T* grad_softmax, T* gradient) { in CSRSparseMatrixSoftmaxGradKernel3D()609 const CSRSparseMatrix& grad_softmax, in CSRSparseMatrixSoftmaxGradGPUImpl()
539 ShapeHandle grad_softmax = sparse_matrix_shape_and_type.shape; in __anon59a9d6c30e02() local
285 def _SoftmaxGrad(op, grad_softmax): argument