convert_sequences_to_tensor_iterator.hpp
1 // Copyright (C) 2018-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 ConvertRNNSequenceToTensorIterator;
18 class TRANSFORMATIONS_API ConvertGRUSequenceToTensorIterator;
19 class TRANSFORMATIONS_API ConvertLSTMSequenceToTensorIterator;
20 
21 } // namespace pass
22 } // namespace ngraph
23 
24 /**
25  * @ingroup ie_transformation_common_api
26  * @brief ConvertRNNSequenceToTensorIterator transformation converts RNNSequence layer to TensorIterator
27  * *
28  */
29 
30 class ngraph::pass::ConvertRNNSequenceToTensorIterator: public ngraph::pass::MatcherPass {
31 public:
32  NGRAPH_RTTI_DECLARATION;
34 };
35 
36 /**
37  * @ingroup ie_transformation_common_api
38  * @brief ConvertGRUSequenceToTensorIterator transformation converts GRUSequence layer to TensorIterator
39  * *
40  */
41 
42 class ngraph::pass::ConvertGRUSequenceToTensorIterator: public ngraph::pass::MatcherPass {
43 public:
44  NGRAPH_RTTI_DECLARATION;
46 };
47 
48 /**
49  * @ingroup ie_transformation_common_api
50  * @brief ConvertLSTMSequenceToTensorIterator transformation converts LSTMSequence layer to TensorIterator
51  * *
52  */
53 
54 class ngraph::pass::ConvertLSTMSequenceToTensorIterator: public ngraph::pass::MatcherPass {
55 public:
56  NGRAPH_RTTI_DECLARATION;
58 };
ConvertGRUSequenceToTensorIterator transformation converts GRUSequence layer to TensorIterator *.
Definition: convert_sequences_to_tensor_iterator.hpp:42
ConvertLSTMSequenceToTensorIterator transformation converts LSTMSequence layer to TensorIterator *.
Definition: convert_sequences_to_tensor_iterator.hpp:54
ConvertRNNSequenceToTensorIterator transformation converts RNNSequence layer to TensorIterator *.
Definition: convert_sequences_to_tensor_iterator.hpp:30
ngraph namespace
Definition: add_fake_quantize_fusion.hpp:14