avg_pool pad fix

apache · Apr 29, 2019 · 6172bde · 6172bde
1 parent fd0c6a3
commit 6172bde
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/python/tvm/relay/frontend/darknet.py b/python/tvm/relay/frontend/darknet.py
@@ -60,9 +60,9 @@ def _darknet_maxpooling(inputs, params, attrs, prefix):
     extra_pad_size = attrs.get('extra_pad_size', 0)
     if extra_pad_size:
         pad_width = ((0, 0), (0, 0), (0, extra_pad_size), (0, extra_pad_size))
-        inputs = get_relay_op('pad')(*inputs,
+        inputs = [get_relay_op('pad')(*inputs,
                                      pad_width=pad_width,
-                                     pad_value=np.finfo(np.float32).min)
+                                     pad_value=np.finfo(np.float32).min)]
     return get_relay_op('max_pool2d')(*inputs, **new_attrs)
 
 def _darknet_avgpooling(inputs, params, attrs, prefix):