Searched refs:grad_softmax (Results 1 – 1 of 1) sorted by relevance
275 def _SoftmaxGrad(op, grad_softmax): argument295 sum_channels = math_ops.reduce_sum(grad_softmax * softmax, -1, keepdims=True)296 return (grad_softmax - sum_channels) * softmax