Skip to content

N
ngraph
Commit c94dd26c authored by tsocha's avatar tsocha Committed by Scott Cyphers
Browse files
Options

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

parent 16d4f547
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
python/ngraph/ops.py
View file @ c94dd26c
...@@ -278,7 +278,7 @@ def avg_pool(x, # type: Node ...@@ -278,7 +278,7 @@ def avg_pool(x, # type: Node
padding_below = [0] * len(window_shape) padding_below = [0] * len(window_shape)
return AvgPool(x, Shape(window_shape), Strides(strides), 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 @nameable_op
...@@ -299,7 +299,7 @@ def max_pool(x, # type: Node ...@@ -299,7 +299,7 @@ def max_pool(x, # type: Node
padding_below = [0] * len(window_shape) padding_below = [0] * len(window_shape)
return MaxPool(x, Shape(window_shape), Strides(strides), return MaxPool(x, Shape(window_shape), Strides(strides),
Shape(padding_above), Shape(padding_above)) Shape(padding_above), Shape(padding_below))
# reduction ops # reduction ops
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment