fix get layout in to_relay (apache#2610)

neo-ai · Feb 20, 2019 · 6b3f945 · 6b3f945
1 parent 1f81681
commit 6b3f945
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(