Implemented cuDNN softmax but it is not currently passing tests.

I added the check for this higher in the function.