From 9fa8ad54047eb4ae515acbebaed8f36fbd65bd2e Mon Sep 17 00:00:00 2001 From: Mikhail Novozhilov Date: Wed, 13 Jan 2021 13:35:55 +0300 Subject: [PATCH] [IE][VPU]: Removed Mish decomposition (#3663) Removed Mish decomposition because new Mish kernel implemented --- .../transformations/mish_decomposition.hpp | 18 ------- .../transformations/mish_decomposition.cpp | 49 ------------------- .../src/frontend/frontend.cpp | 3 -- 3 files changed, 70 deletions(-) delete mode 100644 inference-engine/src/vpu/common/include/vpu/ngraph/transformations/mish_decomposition.hpp delete mode 100644 inference-engine/src/vpu/common/src/ngraph/transformations/mish_decomposition.cpp diff --git a/inference-engine/src/vpu/common/include/vpu/ngraph/transformations/mish_decomposition.hpp b/inference-engine/src/vpu/common/include/vpu/ngraph/transformations/mish_decomposition.hpp deleted file mode 100644 index f9252ca4165..00000000000 --- a/inference-engine/src/vpu/common/include/vpu/ngraph/transformations/mish_decomposition.hpp +++ /dev/null @@ -1,18 +0,0 @@ -// Copyright (C) 2020 Intel Corporation -// SPDX-License-Identifier: Apache-2.0 -// - -#pragma once - -#include - -namespace vpu { - -class MishDecomposition : public ngraph::pass::MatcherPass { -public: - NGRAPH_RTTI_DECLARATION; - MishDecomposition(); -}; - -} // namespace vpu - diff --git a/inference-engine/src/vpu/common/src/ngraph/transformations/mish_decomposition.cpp b/inference-engine/src/vpu/common/src/ngraph/transformations/mish_decomposition.cpp deleted file mode 100644 index fff704d0196..00000000000 --- a/inference-engine/src/vpu/common/src/ngraph/transformations/mish_decomposition.cpp +++ /dev/null @@ -1,49 +0,0 @@ -// Copyright (C) 2020 Intel Corporation -// SPDX-License-Identifier: Apache-2.0 -// - -#include "vpu/ngraph/transformations/mish_decomposition.hpp" - -#include -#include -#include - -#include -#include - -NGRAPH_RTTI_DEFINITION(vpu::MishDecomposition, "MishDecomposition", 0); - -namespace vpu { - -MishDecomposition::MishDecomposition() { - const auto mishPattern = ngraph::pattern::wrap_type(); - - ngraph::matcher_pass_callback callback = [this](ngraph::pattern::Matcher &matcher) { - const auto& mish = ngraph::as_type_ptr(matcher.get_match_root()); - - if (!mish || transformation_callback(mish)) { - return false; - } - - const auto inputType = mish->input_value(0).get_element_type(); - const auto addConst = ngraph::opset5::Constant::create(inputType, ngraph::Shape{}, {1.0f}); - - const auto exp = std::make_shared(mish->input_value(0)); - const auto add = std::make_shared(exp, addConst); - const auto log = std::make_shared(add); - const auto tanh = std::make_shared(log); - const auto mul = std::make_shared(mish->input_value(0), tanh); - - mul->set_friendly_name(mish->get_friendly_name()); - ngraph::copy_runtime_info(mish, {addConst, exp, add, log, tanh, mul}); - ngraph::replace_node(mish, mul); - - return true; - }; - - const auto matcher = std::make_shared(mishPattern, "MishDecomposition"); - register_matcher(matcher, callback); -} - -} // namespace vpu - diff --git a/inference-engine/src/vpu/graph_transformer/src/frontend/frontend.cpp b/inference-engine/src/vpu/graph_transformer/src/frontend/frontend.cpp index 8b07d44e917..10e4dfbfc3d 100644 --- a/inference-engine/src/vpu/graph_transformer/src/frontend/frontend.cpp +++ b/inference-engine/src/vpu/graph_transformer/src/frontend/frontend.cpp @@ -33,7 +33,6 @@ #include #include #include -#include #include #include "vpu/ngraph/transformations/dynamic_to_static_shape.hpp" #include "vpu/ngraph/transformations/eliminate_shapeof_after_dsr.hpp" @@ -185,8 +184,6 @@ ie::ICNNNetwork::Ptr FrontEnd::convertNetwork(ie::ICNNNetwork& network) { manager.register_pass(); manager.register_pass(); manager.register_pass(); - // WA: Mish is not accurate enough. Remove this decomposition when mish is improved - manager.register_pass(); manager.register_pass(); manager.register_pass(); manager.register_pass();