Documentation
Install OpenVINO™
transformations
include
transformations
tensor_iterator_transformations
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