Searched defs:softmax_grad (Results 1 – 2 of 2) sorted by relevance
184 functor::CSRSparseMatrixSoftmaxGrad<Device, T> softmax_grad; in Compute() local
87 auto softmax_grad = op.output(1); in SoftmaxCrossEntropyWithLogitsGrad() local