Imported Upstream version 1.8.0
[platform/core/ml/nnfw.git] / runtime / onert / core / src / ir / OperationDumper.h
1 /*
2  * Copyright (c) 2018 Samsung Electronics Co., Ltd. All Rights Reserved
3  *
4  * Licensed under the Apache License, Version 2.0 (the "License");
5  * you may not use this file except in compliance with the License.
6  * You may obtain a copy of the License at
7  *
8  *      http://www.apache.org/licenses/LICENSE-2.0
9  *
10  * Unless required by applicable law or agreed to in writing, software
11  * distributed under the License is distributed on an "AS IS" BASIS,
12  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13  * See the License for the specific language governing permissions and
14  * limitations under the License.
15  */
16
17 #ifndef __ONERT_OPERATION_DUMPER_H__
18 #define __ONERT_OPERATION_DUMPER_H__
19
20 #include "ir/OperationVisitor.h"
21 #include <string>
22
23 namespace onert
24 {
25 namespace ir
26 {
27
28 class OperationDumper : public OperationVisitor
29 {
30 public:
31   OperationDumper(const std::string &start_msg);
32
33 public:
34   void visit(const operation::Abs &) override;
35   void visit(const operation::Add &node) override;
36   void visit(const operation::ArgMax &) override;
37   void visit(const operation::AvgPool2D &node) override;
38   void visit(const operation::BatchToSpaceND &node) override;
39   void visit(const operation::BroadcastTo &) override;
40   void visit(const operation::Cast &) override;
41   void visit(const operation::Comparison &) override;
42   void visit(const operation::Concat &node) override;
43   void visit(const operation::Conv2D &node) override;
44   void visit(const operation::ConvertFp16ToFp32 &node) override;
45   void visit(const operation::ConvertFp32ToFp16 &node) override;
46   void visit(const operation::Cos &node) override;
47   void visit(const operation::DepthToSpace &) override;
48   void visit(const operation::DepthwiseConv2D &node) override;
49   void visit(const operation::Dequantize &) override;
50   void visit(const operation::Div &) override;
51   void visit(const operation::EmbeddingLookup &) override;
52   void visit(const operation::Exp &) override;
53   void visit(const operation::ExpandDims &) override;
54   void visit(const operation::Floor &) override;
55   void visit(const operation::FullyConnected &node) override;
56   void visit(const operation::Gather &) override;
57   void visit(const operation::HashtableLookup &) override;
58   void visit(const operation::InstanceNorm &) override;
59   void visit(const operation::L2Normalization &) override;
60   void visit(const operation::L2Pool2D &) override;
61   void visit(const operation::LocalResponseNormalization &) override;
62   void visit(const operation::Log &) override;
63   void visit(const operation::LogicalAnd &) override;
64   void visit(const operation::LogicalNot &) override;
65   void visit(const operation::LogicalOr &) override;
66   void visit(const operation::Logistic &) override;
67   void visit(const operation::LSTM &) override;
68   void visit(const operation::MaxPool2D &node) override;
69   void visit(const operation::Mul &) override;
70   void visit(const operation::Neg &) override;
71   void visit(const operation::Pack &) override;
72   void visit(const operation::Pad &) override;
73   void visit(const operation::Permute &node) override;
74   void visit(const operation::Pow &node) override;
75   void visit(const operation::PReLU &) override;
76   void visit(const operation::Range &) override;
77   void visit(const operation::Reduce &) override;
78   void visit(const operation::ReLU &) override;
79   void visit(const operation::ReLU1 &) override;
80   void visit(const operation::ReLU6 &) override;
81   void visit(const operation::Reshape &node) override;
82   void visit(const operation::ResizeBilinear &) override;
83   void visit(const operation::Reverse &) override;
84   void visit(const operation::RNN &) override;
85   void visit(const operation::Round &) override;
86   void visit(const operation::RSQRT &) override;
87   void visit(const operation::Select &node) override;
88   void visit(const operation::Shape &node) override;
89   void visit(const operation::Sin &node) override;
90   void visit(const operation::Softmax &node) override;
91   void visit(const operation::SpaceToBatchND &) override;
92   void visit(const operation::SpaceToDepth &) override;
93   void visit(const operation::Split &) override;
94   void visit(const operation::SQRT &) override;
95   void visit(const operation::SquaredDifference &) override;
96   void visit(const operation::Squeeze &) override;
97   void visit(const operation::Slice &) override;
98   void visit(const operation::StridedSlice &) override;
99   void visit(const operation::StatelessRandomUniform &) override;
100   void visit(const operation::Sub &) override;
101   void visit(const operation::Tanh &) override;
102   void visit(const operation::Tile &) override;
103   void visit(const operation::TopKV2 &) override;
104   void visit(const operation::TransposeConv &) override;
105   void visit(const operation::Transpose &) override;
106   void visit(const operation::Unpack &) override;
107   void visit(const operation::Min &) override;
108   void visit(const operation::Max &) override;
109   void visit(const operation::OneHot &) override;
110   void visit(const operation::If &) override;
111   void visit(const operation::While &) override;
112   void visit(const operation::ZerosLike &) override;
113 };
114
115 } // namespace ir
116 } // namespace onert
117
118 #endif // __ONERT_OPERATION_DUMPER_H__