Commit 5cec2e37 authored by Tomasz Dołbniak's avatar Tomasz Dołbniak Committed by Scott Cyphers

[ONNX] Missing optional output in MaxPool (#2694)

* [ONNX] Use NullNode for the optional output in MaxPool

* Style check
parent 667d8f9e
......@@ -15,6 +15,7 @@
//*****************************************************************************
#include "ngraph/op/max_pool.hpp"
#include "core/null_node.hpp"
#include "ngraph/node.hpp"
#include "ngraph/node_vector.hpp"
#include "utils/convpool.hpp"
......@@ -29,7 +30,9 @@ namespace ngraph
{
NodeVector max_pool(const Node& node)
{
return convpool::make_ng_pool<ngraph::op::MaxPool>(node);
auto max_pool = convpool::make_ng_pool<ngraph::op::MaxPool>(node);
max_pool.emplace_back(std::make_shared<NullNode>()); // Indices (optional)
return max_pool;
}
} // namespace set_1
......
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