From b9f84530daa49ecee7fc98cd0afa3ef92c38487d Mon Sep 17 00:00:00 2001 From: Mateusz Tabaka Date: Mon, 13 Dec 2021 17:30:22 +0100 Subject: [PATCH] Remove identity on MaxPool outputs (#9163) --- src/bindings/python/src/pyopenvino/core/common.cpp | 1 - src/frontends/onnx/frontend/src/utils/pooling_factory.cpp | 7 +------ 2 files changed, 1 insertion(+), 7 deletions(-) diff --git a/src/bindings/python/src/pyopenvino/core/common.cpp b/src/bindings/python/src/pyopenvino/core/common.cpp index 82d337690dc..987456b34ee 100644 --- a/src/bindings/python/src/pyopenvino/core/common.cpp +++ b/src/bindings/python/src/pyopenvino/core/common.cpp @@ -319,7 +319,6 @@ py::dict outputs_to_dict(const std::vector>& outputs, py::dict res; for (const auto& out : outputs) { ov::runtime::Tensor t{request.get_tensor(out)}; - std::cout << t.get_element_type() << " !\n"; switch (t.get_element_type()) { case ov::element::Type_t::i8: { py::array arr(t.get_shape(), t.data()); diff --git a/src/frontends/onnx/frontend/src/utils/pooling_factory.cpp b/src/frontends/onnx/frontend/src/utils/pooling_factory.cpp index 4984698bbac..501a4a6584a 100644 --- a/src/frontends/onnx/frontend/src/utils/pooling_factory.cpp +++ b/src/frontends/onnx/frontend/src/utils/pooling_factory.cpp @@ -28,11 +28,6 @@ std::shared_ptr transposition_axis_order(const Rank& in return std::make_shared(element::i32, Shape{rank}, axes); } - -std::shared_ptr identity(Output node_output) { - const auto zero = default_opset::Constant::create(node_output.get_element_type(), {}, {0}); - return std::make_shared(node_output, zero); -} } // namespace PoolingFactory::PoolingFactory(const Node& node) @@ -89,7 +84,7 @@ OutputVector PoolingFactory::make_max_pool_with_indices() const { return {max_pool->output(0), transposed_indices}; } else { - return {identity(max_pool->output(0)), identity(max_pool->output(1))}; + return {max_pool->output(0), max_pool->output(1)}; } } } // namespace pooling