Commit c7165427 by Yizhi Liu Committed by Tianqi Chen

fix get layout in to_relay (#2610)

parent 18c36ab5
......@@ -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(
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment