Commit ec90481e authored by tsocha's avatar tsocha Committed by Artur Wojcik

[ONNX] Fix bugs in unsqueeze and pooling ops (#1644)

* Unsqueeze Fix

* Pooling fix
parent 02fad9e7
...@@ -37,7 +37,7 @@ namespace ngraph ...@@ -37,7 +37,7 @@ namespace ngraph
ASSERT_VALID_ARGUMENT(node, !axes.empty()) << "'axes' attribute is mandatory."; ASSERT_VALID_ARGUMENT(node, !axes.empty()) << "'axes' attribute is mandatory.";
std::sort(std::begin(axes), std::end(axes), std::greater<int64_t>()); std::sort(std::begin(axes), std::end(axes), std::less<int64_t>());
AxisVector input_order{reshape::get_default_axis_vector(data_shape.size())}; AxisVector input_order{reshape::get_default_axis_vector(data_shape.size())};
......
...@@ -104,8 +104,8 @@ namespace ngraph ...@@ -104,8 +104,8 @@ namespace ngraph
auto paddings = convpool::get_pads(node); auto paddings = convpool::get_pads(node);
// Convert padding from CoordinateDiff to Shape objects // Convert padding from CoordinateDiff to Shape objects
const CoordinateDiff& padding_below{paddings.first}; const CoordinateDiff& padding_above{paddings.first};
const CoordinateDiff& padding_above{paddings.second}; const CoordinateDiff& padding_below{paddings.second};
Shape padding_below_shape{std::begin(padding_below), std::end(padding_below)}; Shape padding_below_shape{std::begin(padding_below), std::end(padding_below)};
Shape padding_above_shape{std::begin(padding_above), std::end(padding_above)}; Shape padding_above_shape{std::begin(padding_above), std::end(padding_above)};
......
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