apply_transformations_to_ti_body.hpp
1 // Copyright (C) 2020 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 #include <ngraph/pass/manager.hpp>
12 #include <ngraph/pass/graph_rewrite.hpp>
13 
14 namespace ngraph {
15 namespace pass {
16 
17 class TRANSFORMATIONS_API ApplyTransformationsToTIBody;
18 
19 } // namespace pass
20 } // namespace ngraph
21 
22 /**
23  * @ingroup ie_transformation_common_api
24  * @brief Applies transformations from given pass::Manager for each TensorIterator body
25  */
26 
27 class ngraph::pass::ApplyTransformationsToTIBody: public ngraph::pass::MatcherPass {
28 public:
29  explicit ApplyTransformationsToTIBody(ngraph::pass::Manager & manager);
30 };
ngraph::pass::ApplyTransformationsToTIBody
Applies transformations from given pass::Manager for each TensorIterator body.
Definition: apply_transformations_to_ti_body.hpp:27
ngraph
ngraph namespace
Definition: batch_norm_decomposition.hpp:17