space_to_batch_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 SpaceToBatchFusion;
18 
19 } // namespace pass
20 } // namespace ngraph
21 
22 /**
23  * @ingroup ie_transformation_common_api
24  * @brief SpaceToBatchFusion transformation replaces following graph:
25  * Transpose (or Reshape) -> Pad -> SpaceToDepth -> Transpose (or Reshape)
26  * to SpaceToBatch
27  * Restrictions:
28  * - input rank must be 4
29  * - Transpose permutation must be [1, 0, 2, 3]
30  * - pad value is 0, PadMode is CONSTANT
31  * - SpaceToDepthMode must be BLOCKS_FIRST
32  */
33 
34 class ngraph::pass::SpaceToBatchFusion: public ngraph::pass::MatcherPass {
35 public:
36  NGRAPH_RTTI_DECLARATION;
38 };
SpaceToBatchFusion transformation replaces following graph: Transpose (or Reshape) -> Pad -> SpaceToD...
Definition: space_to_batch_fusion.hpp:34
ngraph namespace
Definition: add_fake_quantize_fusion.hpp:14