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
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())};
......
......@@ -104,8 +104,8 @@ namespace ngraph
auto paddings = convpool::get_pads(node);
// Convert padding from CoordinateDiff to Shape objects
const CoordinateDiff& padding_below{paddings.first};
const CoordinateDiff& padding_above{paddings.second};
const CoordinateDiff& padding_above{paddings.first};
const CoordinateDiff& padding_below{paddings.second};
Shape padding_below_shape{std::begin(padding_below), std::end(padding_below)};
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