blob: 29a2981463c5d96c7af72433426316953bf75a7f [file] [log] [blame]
Georgios Pinitasd8734b52017-12-22 15:27:52 +00001/*
Giuseppe Rossinibb365de2019-02-15 10:24:47 +00002 * Copyright (c) 2018-2019 ARM Limited.
Georgios Pinitasd8734b52017-12-22 15:27:52 +00003 *
4 * SPDX-License-Identifier: MIT
5 *
6 * Permission is hereby granted, free of charge, to any person obtaining a copy
7 * of this software and associated documentation files (the "Software"), to
8 * deal in the Software without restriction, including without limitation the
9 * rights to use, copy, modify, merge, publish, distribute, sublicense, and/or
10 * sell copies of the Software, and to permit persons to whom the Software is
11 * furnished to do so, subject to the following conditions:
12 *
13 * The above copyright notice and this permission notice shall be included in all
14 * copies or substantial portions of the Software.
15 *
16 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
17 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
18 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
19 * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
20 * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
21 * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
22 * SOFTWARE.
23 */
Georgios Pinitasd9eb2752018-04-03 13:44:29 +010024#ifndef __ARM_COMPUTE_GRAPH_TYPE_PRINTER_H__
25#define __ARM_COMPUTE_GRAPH_TYPE_PRINTER_H__
Georgios Pinitasd8734b52017-12-22 15:27:52 +000026
27#include "arm_compute/core/Error.h"
28#include "arm_compute/core/Types.h"
Georgios Pinitasd9eb2752018-04-03 13:44:29 +010029#include "arm_compute/graph/Types.h"
Georgios Pinitasd8734b52017-12-22 15:27:52 +000030
Georgios Pinitas12be7ab2018-07-03 12:06:23 +010031#include "utils/TypePrinter.h"
32
Georgios Pinitasd8734b52017-12-22 15:27:52 +000033namespace arm_compute
34{
Georgios Pinitasd9eb2752018-04-03 13:44:29 +010035namespace graph
Georgios Pinitasd8734b52017-12-22 15:27:52 +000036{
Georgios Pinitasd8734b52017-12-22 15:27:52 +000037/** Formatted output of the Target. */
38inline ::std::ostream &operator<<(::std::ostream &os, const Target &target)
39{
40 switch(target)
41 {
42 case Target::UNSPECIFIED:
43 os << "UNSPECIFIED";
44 break;
45 case Target::NEON:
46 os << "NEON";
47 break;
48 case Target::CL:
49 os << "CL";
50 break;
Georgios Pinitas2e01e182018-06-06 14:35:15 +010051 case Target::GC:
52 os << "GC";
53 break;
Georgios Pinitasd8734b52017-12-22 15:27:52 +000054 default:
55 ARM_COMPUTE_ERROR("NOT_SUPPORTED!");
56 }
57
58 return os;
59}
60
Georgios Pinitasda2491f2018-06-01 17:49:09 +010061inline ::std::ostream &operator<<(::std::ostream &os, const NodeType &node_type)
62{
63 switch(node_type)
64 {
65 case NodeType::ActivationLayer:
66 os << "ActivationLayer";
67 break;
68 case NodeType::BatchNormalizationLayer:
69 os << "BatchNormalizationLayer";
70 break;
Manuel Bottinid2048ce2018-10-23 17:00:42 +010071 case NodeType::BoundingBoxTransformLayer:
72 os << "BoundingBoxTransformLayer";
73 break;
Georgios Pinitasda2491f2018-06-01 17:49:09 +010074 case NodeType::ChannelShuffleLayer:
75 os << "ChannelShuffleLayer";
76 break;
Georgios Pinitase2220552018-07-20 13:23:44 +010077 case NodeType::ConcatenateLayer:
78 os << "ConcatenateLayer";
79 break;
Georgios Pinitasda2491f2018-06-01 17:49:09 +010080 case NodeType::ConvolutionLayer:
81 os << "ConvolutionLayer";
82 break;
83 case NodeType::DeconvolutionLayer:
84 os << "DeconvolutionLayer";
85 break;
Isabella Gottardi7234ed82018-11-27 08:51:10 +000086 case NodeType::DetectionOutputLayer:
87 os << "DetectionOutputLayer";
88 break;
Georgios Pinitasda2491f2018-06-01 17:49:09 +010089 case NodeType::DepthwiseConvolutionLayer:
90 os << "DepthwiseConvolutionLayer";
91 break;
92 case NodeType::EltwiseLayer:
93 os << "EltwiseLayer";
94 break;
95 case NodeType::FlattenLayer:
96 os << "FlattenLayer";
97 break;
98 case NodeType::FullyConnectedLayer:
99 os << "FullyConnectedLayer";
100 break;
giuros01acce5042019-02-21 17:32:34 +0000101 case NodeType::FusedConvolutionBatchNormalizationLayer:
102 os << "FusedConvolutionBatchNormalizationLayer";
103 break;
Manuel Bottini5209be52019-02-13 16:34:56 +0000104 case NodeType::GenerateProposalsLayer:
105 os << "GenerateProposalsLayer";
106 break;
Georgios Pinitasda2491f2018-06-01 17:49:09 +0100107 case NodeType::NormalizationLayer:
108 os << "NormalizationLayer";
109 break;
Michele Di Giorgio555d1102018-09-12 13:51:59 +0100110 case NodeType::NormalizePlanarYUVLayer:
111 os << "NormalizePlanarYUVLayer";
112 break;
Michele Di Giorgio4bb17332018-09-26 13:56:51 +0100113 case NodeType::PadLayer:
114 os << "PadLayer";
115 break;
Georgios Pinitas57c48242018-08-02 13:41:49 +0100116 case NodeType::PermuteLayer:
117 os << "PermuteLayer";
118 break;
Georgios Pinitasda2491f2018-06-01 17:49:09 +0100119 case NodeType::PoolingLayer:
120 os << "PoolingLayer";
121 break;
Pablo Tello32521432018-11-15 14:43:10 +0000122 case NodeType::PriorBoxLayer:
123 os << "PriorBoxLayer";
124 break;
Michele Di Giorgio555d1102018-09-12 13:51:59 +0100125 case NodeType::ReorgLayer:
126 os << "ReorgLayer";
127 break;
Georgios Pinitasda2491f2018-06-01 17:49:09 +0100128 case NodeType::ReshapeLayer:
129 os << "ReshapeLayer";
130 break;
131 case NodeType::ResizeLayer:
132 os << "ResizeLayer";
133 break;
Manuel Bottini3f9d4d72018-10-19 14:04:42 +0100134 case NodeType::ROIAlignLayer:
135 os << "ROIAlignLayer";
136 break;
Georgios Pinitasda2491f2018-06-01 17:49:09 +0100137 case NodeType::SoftmaxLayer:
138 os << "SoftmaxLayer";
139 break;
Michele Di Giorgioc30b6682018-09-12 17:44:08 +0100140 case NodeType::SliceLayer:
141 os << "SliceLayer";
142 break;
Georgios Pinitasda2491f2018-06-01 17:49:09 +0100143 case NodeType::SplitLayer:
144 os << "SplitLayer";
145 break;
Michele Di Giorgioec699752019-03-22 15:25:32 +0000146 case NodeType::StackLayer:
147 os << "StackLayer";
148 break;
Michalis Spyrou4e1c3f32018-09-20 17:14:03 +0100149 case NodeType::UpsampleLayer:
150 os << "UpsampleLayer";
151 break;
Michalis Spyrou96f67692018-09-13 11:39:28 +0100152 case NodeType::YOLOLayer:
153 os << "YOLOLayer";
154 break;
Georgios Pinitasda2491f2018-06-01 17:49:09 +0100155 case NodeType::Input:
156 os << "Input";
157 break;
158 case NodeType::Output:
159 os << "Output";
160 break;
161 case NodeType::Const:
162 os << "Const";
163 break;
164 case NodeType::Dummy:
165 os << "Dummy";
166 break;
167 default:
168 ARM_COMPUTE_ERROR("NOT_SUPPORTED!");
169 }
170
171 return os;
172}
173
Georgios Pinitasd8734b52017-12-22 15:27:52 +0000174/** Formatted output of the EltwiseOperation type. */
175inline ::std::ostream &operator<<(::std::ostream &os, const EltwiseOperation &eltwise_op)
176{
177 switch(eltwise_op)
178 {
Georgios Pinitase2220552018-07-20 13:23:44 +0100179 case EltwiseOperation::Add:
180 os << "Add";
Georgios Pinitasd8734b52017-12-22 15:27:52 +0000181 break;
Georgios Pinitase2220552018-07-20 13:23:44 +0100182 case EltwiseOperation::Mul:
183 os << "Mul";
Georgios Pinitasd8734b52017-12-22 15:27:52 +0000184 break;
Georgios Pinitase2220552018-07-20 13:23:44 +0100185 case EltwiseOperation::Sub:
186 os << "Sub";
Georgios Pinitasd8734b52017-12-22 15:27:52 +0000187 break;
188 default:
189 ARM_COMPUTE_ERROR("NOT_SUPPORTED!");
190 }
191
192 return os;
193}
194
195/** Formatted output of the ConvolutionMethod type. */
196inline ::std::ostream &operator<<(::std::ostream &os, const ConvolutionMethod &method)
197{
198 switch(method)
199 {
Georgios Pinitase2220552018-07-20 13:23:44 +0100200 case ConvolutionMethod::Default:
201 os << "Default";
Georgios Pinitasd8734b52017-12-22 15:27:52 +0000202 break;
Georgios Pinitase2220552018-07-20 13:23:44 +0100203 case ConvolutionMethod::Direct:
204 os << "Direct";
Georgios Pinitasd8734b52017-12-22 15:27:52 +0000205 break;
206 case ConvolutionMethod::GEMM:
207 os << "GEMM";
208 break;
Georgios Pinitase2220552018-07-20 13:23:44 +0100209 case ConvolutionMethod::Winograd:
210 os << "Winograd";
Georgios Pinitasd8734b52017-12-22 15:27:52 +0000211 break;
212 default:
213 ARM_COMPUTE_ERROR("NOT_SUPPORTED!");
214 }
215
216 return os;
217}
218
Giorgio Arena59631a12018-05-02 13:59:04 +0100219/** Formatted output of the FastMathHint type. */
220inline ::std::ostream &operator<<(::std::ostream &os, const FastMathHint &hint)
221{
222 switch(hint)
223 {
Georgios Pinitase2220552018-07-20 13:23:44 +0100224 case FastMathHint::Enabled:
225 os << "Enabled";
Giorgio Arena59631a12018-05-02 13:59:04 +0100226 break;
Georgios Pinitase2220552018-07-20 13:23:44 +0100227 case FastMathHint::Disabled:
228 os << "Disabled";
Giorgio Arena59631a12018-05-02 13:59:04 +0100229 break;
230 default:
231 ARM_COMPUTE_ERROR("NOT_SUPPORTED!");
232 }
233
234 return os;
235}
236
Georgios Pinitasd8734b52017-12-22 15:27:52 +0000237/** Formatted output of the DepthwiseConvolutionMethod type. */
238inline ::std::ostream &operator<<(::std::ostream &os, const DepthwiseConvolutionMethod &method)
239{
240 switch(method)
241 {
Georgios Pinitase2220552018-07-20 13:23:44 +0100242 case DepthwiseConvolutionMethod::Default:
Georgios Pinitasd8734b52017-12-22 15:27:52 +0000243 os << "DEFAULT";
244 break;
245 case DepthwiseConvolutionMethod::GEMV:
246 os << "GEMV";
247 break;
Georgios Pinitase2220552018-07-20 13:23:44 +0100248 case DepthwiseConvolutionMethod::Optimized3x3:
249 os << "Optimized3x3";
Georgios Pinitasd8734b52017-12-22 15:27:52 +0000250 break;
251 default:
252 ARM_COMPUTE_ERROR("NOT_SUPPORTED!");
253 }
254
255 return os;
256}
Georgios Pinitasd9eb2752018-04-03 13:44:29 +0100257} // namespace graph
Georgios Pinitasd8734b52017-12-22 15:27:52 +0000258} // namespace arm_compute
Georgios Pinitasd9eb2752018-04-03 13:44:29 +0100259#endif /* __ARM_COMPUTE_GRAPH_TYPE_PRINTER_H__ */