batch_to_space_fusion.hpp
1 // Copyright (C) 2021 Intel Corporation
2 // SPDX-License-Identifier: Apache-2.0
3 //
4 
5 #pragma once
6 
7 #include <vector>
8 #include <memory>
9 
10 #include <transformations_visibility.hpp>
11 
12 #include <ngraph/pass/graph_rewrite.hpp>
13 
14 namespace ngraph {
15 namespace pass {
16 
17 class TRANSFORMATIONS_API BatchToSpaceFusion;
18 
19 } // namespace pass
20 } // namespace ngraph
21 
22 /**
23  * @ingroup ie_transformation_common_api
24  * @brief BatchToSpaceFusion transformation replaces following graph:
25  * Transpose (or Reshape) -> DepthToSpace -> StridedSlice -> Transpose (or Reshape)
26  * to BatchToSpace
27  * Restrictions:
28  * - input rank must be 4
29  * - Transpose permutation must be [1, 0, 2, 3]
30  * - DepthToSpaceMode must be BLOCKS_FIRST
31  */
32 
33 class ngraph::pass::BatchToSpaceFusion: public ngraph::pass::MatcherPass {
34 public:
35  NGRAPH_RTTI_DECLARATION;
37 };
BatchToSpaceFusion transformation replaces following graph: Transpose (or Reshape) -> DepthToSpace ->...
Definition: batch_to_space_fusion.hpp:33
ngraph namespace
Definition: add_fake_quantize_fusion.hpp:14