From 1e9f642ba22734664c1c28dc452fe62e09b84051 Mon Sep 17 00:00:00 2001 From: "Steven S. Lyubomirsky" Date: Mon, 7 Dec 2020 17:37:38 -0800 Subject: [PATCH] Linting fix --- python/tvm/relay/frontend/keras.py | 6 ++---- 1 file changed, 2 insertions(+), 4 deletions(-) diff --git a/python/tvm/relay/frontend/keras.py b/python/tvm/relay/frontend/keras.py index f6e3f10e3d90..1c4606541559 100644 --- a/python/tvm/relay/frontend/keras.py +++ b/python/tvm/relay/frontend/keras.py @@ -632,13 +632,11 @@ def _convert_pooling( if pool_type == "GlobalMaxPooling2D": return _convert_flatten( - _op.nn.global_max_pool2d(inexpr, **global_pool_params), - keras_layer, etab, data_layout + _op.nn.global_max_pool2d(inexpr, **global_pool_params), keras_layer, etab, data_layout ) if pool_type == "GlobalAveragePooling2D": return _convert_flatten( - _op.nn.global_avg_pool2d(inexpr, **global_pool_params), - keras_layer, etab, data_layout + _op.nn.global_avg_pool2d(inexpr, **global_pool_params), keras_layer, etab, data_layout ) pool_h, pool_w = keras_layer.pool_size stride_h, stride_w = keras_layer.strides