gather_normalize_negative_indices.hpp
1 // Copyright (C) 2021 Intel Corporation
2 // SPDX-License-Identifier: Apache-2.0
3 //
4 
5 #pragma once
6 
7 #include <transformations_visibility.hpp>
8 #include <ngraph/pass/graph_rewrite.hpp>
9 
10 namespace ngraph {
11 namespace pass {
12 
13  class TRANSFORMATIONS_API GatherNegativeConstIndicesNormalize;
14 
15 } // namespace pass
16 } // namespace ngraph
17 
18 /**
19  * @ingroup ie_transformation_common_api
20  * @brief GatherNegativeConstIndicesNormalize checks if indices value is negative scalar and
21  * normalizes it using ShapeOf->Add->Cast subgraph.
22  * We need to remove this transformation after adding support of negative indices in
23  * future version of Gather operation.
24  */
25 class ngraph::pass::GatherNegativeConstIndicesNormalize : public ngraph::pass::MatcherPass {
26 public:
27  NGRAPH_RTTI_DECLARATION;
29 };
GatherNegativeConstIndicesNormalize checks if indices value is negative scalar and normalizes it usin...
Definition: gather_normalize_negative_indices.hpp:25
ngraph namespace
Definition: add_fake_quantize_fusion.hpp:14