[CRASH] cudnn softmax r1
small code refactoring
Merge pull request #2357 from nouiz/cudnn_repair_r1

[CRASH] Cudnn repair r1 softmax