broadcast_elementwise_fusion.hpp
1 // Copyright (C) 2018-2021 Intel Corporation
2 // SPDX-License-Identifier: Apache-2.0
3 //
4 
5 #pragma once
6 
7 #include <transformations_visibility.hpp>
8 
9 #include <ngraph/ngraph.hpp>
10 #include <ngraph/pass/graph_rewrite.hpp>
11 #include "ngraph/pattern/matcher.hpp"
12 
13 namespace ngraph {
14 namespace pass {
15 
16 class TRANSFORMATIONS_API BroadcastElementwiseFusion;
17 
18 } // namespace pass
19 } // namespace ngraph
20 
21 /**
22  * @ingroup ie_transformation_common_api
23  * @brief Removing Broadcast OP before ElementWise if output shape of Broadcast
24  * are equal neighboring input shape of ElementWise.
25  */
26 
27 class ngraph::pass::BroadcastElementwiseFusion: public ngraph::pass::MatcherPass {
28 public:
29  NGRAPH_RTTI_DECLARATION;
31 };
Removing Broadcast OP before ElementWise if output shape of Broadcast are equal neighboring input sha...
Definition: broadcast_elementwise_fusion.hpp:27
ngraph namespace
Definition: add_fake_quantize_fusion.hpp:14