Implemented infer_shape on cudnn ops.