(cherry picked from commit c71654277d)
This commit is contained in:
ZihengJiang 2019-02-18 14:28:48 -08:00
Родитель 48f8198a5b
Коммит 2e5ca24c76
1 изменённых файлов: 1 добавлений и 1 удалений

Просмотреть файл

@ -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(