diff options
Diffstat (limited to 'arm_compute/graph/TypePrinter.h')
| -rw-r--r-- | arm_compute/graph/TypePrinter.h | 6 |
1 files changed, 6 insertions, 0 deletions
diff --git a/arm_compute/graph/TypePrinter.h b/arm_compute/graph/TypePrinter.h index 9117864130..4628d924dc 100644 --- a/arm_compute/graph/TypePrinter.h +++ b/arm_compute/graph/TypePrinter.h @@ -95,12 +95,18 @@ inline ::std::ostream &operator<<(::std::ostream &os, const NodeType &node_type) case NodeType::NormalizationLayer: os << "NormalizationLayer"; break; + case NodeType::NormalizePlanarYUVLayer: + os << "NormalizePlanarYUVLayer"; + break; case NodeType::PermuteLayer: os << "PermuteLayer"; break; case NodeType::PoolingLayer: os << "PoolingLayer"; break; + case NodeType::ReorgLayer: + os << "ReorgLayer"; + break; case NodeType::ReshapeLayer: os << "ReshapeLayer"; break; |