fix get layout in to_relay (#2610)

apache · Feb 16, 2019 · c716542 · c716542
1 parent 18c36ab
commit c716542
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/nnvm/python/nnvm/to_relay.py b/nnvm/python/nnvm/to_relay.py
@@ -127,7 +127,7 @@ def _max_pool2d(children, attrs, odtype='float32'):
     pool_size = attrs.get_int_tuple('pool_size', (1, 1))
     strides = attrs.get_int_tuple('strides', (1, 1))
     padding = attrs.get_int_tuple('padding', (0, 0))
-    layout = attrs.get_int_tuple('layout', 'NCHW')
+    layout = attrs.get_str('layout', 'NCHW')
     ceil_mode = attrs.get_bool('ceil_mode', False)
 
     return op.nn.max_pool2d(