Searched defs:grad_softmax (Results 1 – 5 of 5) sorted by relevance
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()
200 def _SparseMatrixSoftmaxGrad(op, grad_softmax): argument
539 ShapeHandle grad_softmax = sparse_matrix_shape_and_type.shape; in __anona88575120e02() local
93 Tensor grad_softmax; in _flash_attention_backward() local
281 def _SoftmaxGrad(op, grad_softmax): argument