diff options
Diffstat (limited to 'arm_compute/graph/TypePrinter.h')
| -rw-r--r-- | arm_compute/graph/TypePrinter.h | 50 |
1 files changed, 34 insertions, 16 deletions
diff --git a/arm_compute/graph/TypePrinter.h b/arm_compute/graph/TypePrinter.h index d56407b52b..5e83820ab3 100644 --- a/arm_compute/graph/TypePrinter.h +++ b/arm_compute/graph/TypePrinter.h @@ -1,5 +1,5 @@ /* - * Copyright (c) 2018-2020 ARM Limited. + * Copyright (c) 2018-2021, 2023 Arm Limited. * * SPDX-License-Identifier: MIT * @@ -21,8 +21,8 @@ * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE * SOFTWARE. */ -#ifndef ARM_COMPUTE_GRAPH_TYPE_PRINTER_H -#define ARM_COMPUTE_GRAPH_TYPE_PRINTER_H +#ifndef ACL_ARM_COMPUTE_GRAPH_TYPEPRINTER_H +#define ACL_ARM_COMPUTE_GRAPH_TYPEPRINTER_H #include "arm_compute/core/Error.h" #include "arm_compute/core/Types.h" @@ -37,19 +37,19 @@ namespace graph /** Formatted output of the Target. */ inline ::std::ostream &operator<<(::std::ostream &os, const Target &target) { - switch(target) + switch (target) { case Target::UNSPECIFIED: os << "UNSPECIFIED"; break; case Target::NEON: - os << "NEON"; + os << "Neon"; break; case Target::CL: os << "CL"; break; - case Target::GC: - os << "GC"; + case Target::CLVK: + os << "CLVK"; break; default: ARM_COMPUTE_ERROR("NOT_SUPPORTED!"); @@ -60,11 +60,14 @@ inline ::std::ostream &operator<<(::std::ostream &os, const Target &target) inline ::std::ostream &operator<<(::std::ostream &os, const NodeType &node_type) { - switch(node_type) + switch (node_type) { case NodeType::ActivationLayer: os << "ActivationLayer"; break; + case NodeType::ArgMinMaxLayer: + os << "ArgMinMaxLayer"; + break; case NodeType::BatchNormalizationLayer: os << "BatchNormalizationLayer"; break; @@ -83,6 +86,9 @@ inline ::std::ostream &operator<<(::std::ostream &os, const NodeType &node_type) case NodeType::DeconvolutionLayer: os << "DeconvolutionLayer"; break; + case NodeType::DepthToSpaceLayer: + os << "DepthToSpaceLayer"; + break; case NodeType::DequantizationLayer: os << "DequantizationLayer"; break; @@ -98,6 +104,9 @@ inline ::std::ostream &operator<<(::std::ostream &os, const NodeType &node_type) case NodeType::EltwiseLayer: os << "EltwiseLayer"; break; + case NodeType::UnaryEltwiseLayer: + os << "UnaryEltwiseLayer"; + break; case NodeType::FlattenLayer: os << "FlattenLayer"; break; @@ -113,6 +122,9 @@ inline ::std::ostream &operator<<(::std::ostream &os, const NodeType &node_type) case NodeType::GenerateProposalsLayer: os << "GenerateProposalsLayer"; break; + case NodeType::L2NormalizeLayer: + os << "L2NormalizeLayer"; + break; case NodeType::NormalizationLayer: os << "NormalizationLayer"; break; @@ -140,6 +152,9 @@ inline ::std::ostream &operator<<(::std::ostream &os, const NodeType &node_type) case NodeType::QuantizationLayer: os << "QuantizationLayer"; break; + case NodeType::ReductionOperationLayer: + os << "ReductionOperationLayer"; + break; case NodeType::ReorgLayer: os << "ReorgLayer"; break; @@ -164,12 +179,12 @@ inline ::std::ostream &operator<<(::std::ostream &os, const NodeType &node_type) case NodeType::StackLayer: os << "StackLayer"; break; + case NodeType::StridedSliceLayer: + os << "StridedSliceLayer"; + break; case NodeType::UpsampleLayer: os << "UpsampleLayer"; break; - case NodeType::YOLOLayer: - os << "YOLOLayer"; - break; case NodeType::Input: os << "Input"; break; @@ -192,7 +207,7 @@ inline ::std::ostream &operator<<(::std::ostream &os, const NodeType &node_type) /** Formatted output of the EltwiseOperation type. */ inline ::std::ostream &operator<<(::std::ostream &os, const EltwiseOperation &eltwise_op) { - switch(eltwise_op) + switch (eltwise_op) { case EltwiseOperation::Add: os << "Add"; @@ -203,6 +218,9 @@ inline ::std::ostream &operator<<(::std::ostream &os, const EltwiseOperation &el case EltwiseOperation::Sub: os << "Sub"; break; + case EltwiseOperation::Div: + os << "Div"; + break; default: ARM_COMPUTE_ERROR("NOT_SUPPORTED!"); } @@ -213,7 +231,7 @@ inline ::std::ostream &operator<<(::std::ostream &os, const EltwiseOperation &el /** Formatted output of the ConvolutionMethod type. */ inline ::std::ostream &operator<<(::std::ostream &os, const ConvolutionMethod &method) { - switch(method) + switch (method) { case ConvolutionMethod::Default: os << "Default"; @@ -237,7 +255,7 @@ inline ::std::ostream &operator<<(::std::ostream &os, const ConvolutionMethod &m /** Formatted output of the FastMathHint type. */ inline ::std::ostream &operator<<(::std::ostream &os, const FastMathHint &hint) { - switch(hint) + switch (hint) { case FastMathHint::Enabled: os << "Enabled"; @@ -255,7 +273,7 @@ inline ::std::ostream &operator<<(::std::ostream &os, const FastMathHint &hint) /** Formatted output of the DepthwiseConvolutionMethod type. */ inline ::std::ostream &operator<<(::std::ostream &os, const DepthwiseConvolutionMethod &method) { - switch(method) + switch (method) { case DepthwiseConvolutionMethod::Default: os << "DEFAULT"; @@ -271,4 +289,4 @@ inline ::std::ostream &operator<<(::std::ostream &os, const DepthwiseConvolution } } // namespace graph } // namespace arm_compute -#endif /* ARM_COMPUTE_GRAPH_TYPE_PRINTER_H */ +#endif // ACL_ARM_COMPUTE_GRAPH_TYPEPRINTER_H |