Searched defs:grad_softmax (Results 1 – 4 of 4) sorted by relevance
78 def _SparseMatrixSoftmaxGrad(op, grad_softmax): argument
495 const int* grad_softmax_col_ind, const T* grad_softmax, T* gradient) { in CalculateRowSoftmaxGrad()561 const int* grad_softmax_col_ind, const T* grad_softmax, T* gradient) { in CSRSparseMatrixSoftmaxGradKernel2D()582 const T* grad_softmax, T* gradient) { in CSRSparseMatrixSoftmaxGradKernel3D()622 const CSRSparseMatrix& grad_softmax, in CSRSparseMatrixSoftmaxGradGPUImpl()
539 ShapeHandle grad_softmax = sparse_matrix_shape_and_type.shape; in __anond96a6a670e02() local
283 def _SoftmaxGrad(op, grad_softmax): argument