[ONNX] Squeeze op (#1625)

* [ONNX] Squeeze op

* Review fix pt. 1

* Type change

src/ngraph/frontend/onnx_import/CMakeLists.txt

@@ -104,6 +104,8 @@ add_library(onnx_import STATIC
         op/split.cpp
         op/split.hpp
         op/sqrt.hpp
+        op/squeeze.cpp
+        op/squeeze.hpp
         op/sub.hpp
         op/sum.hpp
         op/tanh.hpp

src/ngraph/frontend/onnx_import/op/squeeze.cpp

+//*****************************************************************************
+// Copyright 2018 Intel Corporation
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//     http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+//*****************************************************************************
+
+#include <numeric>
+#include <vector>
+
+#include "ngraph/axis_vector.hpp"
+#include "ngraph/op/reshape.hpp"
+#include "utils/reshape.hpp"
+
+#include "exceptions.hpp"
+#include "squeeze.hpp"
+
+namespace ngraph
+{
+    namespace onnx_import
+    {
+        namespace op
+        {
+            NodeVector squeeze(const Node& node)
+            {
+                NodeVector inputs{node.get_ng_inputs()};
+                auto data = inputs.at(0);
+                auto data_shape = data->get_shape();
+                auto axes = node.get_attribute_value<std::vector<uint64_t>>("axes", {});
+
+                if (axes.empty())
+                {
+                    for (auto index = 0; index < data_shape.size(); ++index)
+                    {
+                        if (data_shape.at(index) == 1)
+                        {
+                            axes.push_back(index);
+                        }
+                    }
+                }
+
+                std::sort(std::begin(axes), std::end(axes), std::greater<uint64_t>());
+
+                AxisVector input_order{reshape::get_default_axis_vector(data_shape.size())};
+
+                for (auto axis : axes)
+                {
+                    data_shape.erase(std::next(std::begin(data_shape), axis));
+                }
+
+                return {std::make_shared<ngraph::op::Reshape>(data, input_order, data_shape)};
+            }
+
+        } // namespace  op
+
+    } // namespace  onnx_import
+
+} // namespace  ngraph

src/ngraph/frontend/onnx_import/op/squeeze.hpp

+//*****************************************************************************
+// Copyright 2018 Intel Corporation
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//     http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+//*****************************************************************************
+
+#pragma once
+
+#include "ngraph/node_vector.hpp"
+
+#include "core/node.hpp"
+
+namespace ngraph
+{
+    namespace onnx_import
+    {
+        namespace op
+        {
+            NodeVector squeeze(const Node& node);
+        } // namespace  op
+
+    } // namespace  onnx_import
+
+} // namespace  ngraph

src/ngraph/frontend/onnx_import/ops_bridge.cpp

@@ -67,6 +67,7 @@
 #include "op/softsign.hpp"
 #include "op/split.hpp"
 #include "op/sqrt.hpp"
+#include "op/squeeze.hpp"
 #include "op/sub.hpp"
 #include "op/sum.hpp"
 #include "op/tanh.hpp"
@@ -182,6 +183,7 @@ namespace ngraph
                     m_map.emplace("Softsign", std::bind(op::softsign, std::placeholders::_1));
                     m_map.emplace("Split", std::bind(op::split, std::placeholders::_1));
                     m_map.emplace("Sqrt", std::bind(op::sqrt, std::placeholders::_1));
+                    m_map.emplace("Squeeze", std::bind(op::squeeze, std::placeholders::_1));
                     m_map.emplace("Sub", std::bind(op::sub, std::placeholders::_1));
                     m_map.emplace("Sum", std::bind(op::sum, std::placeholders::_1));
                     m_map.emplace("Tanh", std::bind(op::tanh, std::placeholders::_1));