onnxruntime

53307a5f - improve perf for softmax (#6128)

improve perf for softmax (#6128) * improve perf for both gathergrad and softmax * revert the change in gathergrad and will be done in another PR. * address comments from code review.