Implemented cuDNN softmax but it is not currently passing tests.

You need to add a test for the optimization. It make sure the optimization get done and that the result is still good.