Forgot to set up data layout in _convert_activation

apache · Dec 8, 2020 · cfc798c · cfc798c
1 parent 29acdba
commit cfc798c
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/python/tvm/relay/frontend/keras.py b/python/tvm/relay/frontend/keras.py
@@ -65,8 +65,10 @@ def _convert_recurrent_activation(inexpr, keras_layer):
 
 
 def _convert_activation(
-    inexpr, keras_layer, _, input_shape=None, data_layout=None
+    inexpr, keras_layer, etab, input_shape=None, data_layout=None
 ):  # pylint: disable=unused-argument
+    if data_layout is None:
+        data_layout = etab.data_layout
     if isinstance(keras_layer, str):
         act_type = keras_layer
     else: