Fixed problem with cudnn softmax tests. This fix should be generalized for use of the class. Also there are two cases it cannot handle.