Skip to content

Commit e0ef163

Browse files
committed
Ensure kern_shape has dtype attribute.
1 parent e500a27 commit e0ef163

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

theano/gpuarray/dnn.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -385,6 +385,7 @@ def __init__(self, border_mode, subsample=(1, 1), conv_mode='conv',
385385
self.precision = precision
386386

387387
def make_node(self, kern_shape):
388+
kern_shape = as_tensor_variable(kern_shape)
388389
if kern_shape.type.ndim != 1 or kern_shape.dtype not in theano.tensor.basic.int_dtypes:
389390
raise TypeError('kern must be an int64 1D shape tensor')
390391
kern_shape = theano.tensor.basic.cast(kern_shape, 'int64')

0 commit comments

Comments
 (0)