From 2e5ca24c769799bbef359d59eb4318f4baec52eb Mon Sep 17 00:00:00 2001 From: ZihengJiang Date: Mon, 18 Feb 2019 14:28:48 -0800 Subject: [PATCH] fix get layout in to_relay (#2610) (cherry picked from commit c71654277deea4e3467d1ff0235c192e8e7113e3) --- nnvm/python/nnvm/to_relay.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/nnvm/python/nnvm/to_relay.py b/nnvm/python/nnvm/to_relay.py index 8ad32821740a..030fe9991331 100644 --- 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(