Implemented cuDNN softmax but it is not currently passing the tests.