Implemented cuDNN softmax but it is not currently passing tests.