Forgot to set data_layout argument

apache · Jan 21, 2022 · d50375d · d50375d
1 parent 1e9f642
commit d50375d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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