From d50375dd6f31cbd6053a47964b0ab03fa38650ab Mon Sep 17 00:00:00 2001 From: "Steven S. Lyubomirsky" Date: Mon, 7 Dec 2020 17:42:46 -0800 Subject: [PATCH] Forgot to set data_layout argument --- python/tvm/relay/frontend/keras.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/python/tvm/relay/frontend/keras.py b/python/tvm/relay/frontend/keras.py index 1c4606541559..ac157b80cf7b 100644 --- a/python/tvm/relay/frontend/keras.py +++ b/python/tvm/relay/frontend/keras.py @@ -353,7 +353,7 @@ def _convert_convolution1d(inexpr, keras_layer, etab, data_layout, input_shape=N else: act_type = keras_layer.activation.__name__ if act_type != "linear": - out = _convert_activation(out, act_type, etab) + out = _convert_activation(out, act_type, etab, data_layout) return out