Fix of argument of padding_below being ignored. (#694)

c94dd26c · tsocha · Scott Cyphers · 16d4f547 · c94dd26c
Commit c94dd26c authored Mar 20, 2018 by tsocha Committed by Scott Cyphers Mar 20, 2018
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 2 deletions

ops.py python/ngraph/ops.py +2 -2

No files found.
--- a/python/ngraph/ops.py
+++ b/python/ngraph/ops.py
@@ -278,7 +278,7 @@ def avg_pool(x,                      # type: Node
        padding_below = [0] * len(window_shape)

    return AvgPool(x, Shape(window_shape), Strides(strides),
-                   Shape(padding_above), Shape(padding_above), zero_pad)
+                   Shape(padding_above), Shape(padding_below), zero_pad)


 @nameable_op
@@ -299,7 +299,7 @@ def max_pool(x,                      # type: Node
        padding_below = [0] * len(window_shape)

    return MaxPool(x, Shape(window_shape), Strides(strides),
-                   Shape(padding_above), Shape(padding_above))
+                   Shape(padding_above), Shape(padding_below))


 # reduction ops