Rename the bitfield to c_axis and change axis to the list of axes from

theano/gpuarray/dnn.py

@@ -1570,7 +1570,7 @@ class GpuDnnReduction(DnnBase):
 
     params_type = ParamsType(red_op=cudnn.cudnnReduceTensorOp_t,
                              acc_dtype=cudnn.cudnnDataType_t,
-                             axis=uint32_t,
+                             c_axis=uint32_t,
                              handle=handle_type)
 
     def __init__(self, red_op, axis, acc_dtype, dtype, arg):
@@ -1587,7 +1587,11 @@ class GpuDnnReduction(DnnBase):
                 raise ValueError('Too many axes to reduce on')
             if any(a >= 8 for a in axis):
                 raise ValueError('Axes larger than 8 not supported')
-        self.axis = self._convert_axis(axis)
+            axis = tuple(axis)
+        # c_axis is a bitfield (1 to reduce)
+        self.c_axis = self._convert_axis(axis)
+        # axis is a list of axes to reduce on
+        self.axis = axis
         if arg and (red_op != 'max' and red_op != 'min'):
             raise ValueError("Can't request indices for something other than min or max")
         self.arg = arg

theano/gpuarray/dnn_redux.c

@@ -72,7 +72,7 @@ int APPLY_SPECIFIC(dnn_redux)(PyGpuArrayObject *input,
   p = 0;
   rsz = 1;
   for (unsigned int i = 0; i < PyGpuArray_NDIM(input); i++) {
-    if (!(params->axis & (1U << i))) {
+    if (!(params->c_axis & (1U << i))) {
       dims[p] = PyGpuArray_DIM(input, i);
       p++;
     } else {
@@ -111,7 +111,7 @@ int APPLY_SPECIFIC(dnn_redux)(PyGpuArrayObject *input,
   // We have to do some trickery to be able to pass it what it need.
   p = 0;
   for (unsigned int i = 0; i < PyGpuArray_NDIM(input); i++) {
-    if (params->axis & (1U << i)) {
+    if (params->c_axis & (1U << i)) {
       dims[i] = 1;
       strs[i] = 0;
     } else {