It makes no sense to have a softmax_grad node with inputs that are not
matrices of the same size. C implementation would have complained.
T_SoftmaxGrad.test_infer_shape now passes in FAST_COMPILE.