From c9fe9fc3fc8c58a9d724776f75831ab35f07b253 Mon Sep 17 00:00:00 2001 From: Giorgio Arena Date: Wed, 6 Oct 2021 12:54:29 +0100 Subject: Add support for 5D data layout indexing Signed-off-by: Giorgio Arena Change-Id: Ib346bb6b90d2220ec5934c83a9a1f0cd540b8731 Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/6377 Comments-Addressed: Arm Jenkins Reviewed-by: Gunes Bayir Tested-by: Arm Jenkins --- utils/TypePrinter.h | 6 ++++++ 1 file changed, 6 insertions(+) (limited to 'utils') diff --git a/utils/TypePrinter.h b/utils/TypePrinter.h index 5fa92e6360..0bea408519 100644 --- a/utils/TypePrinter.h +++ b/utils/TypePrinter.h @@ -666,6 +666,9 @@ inline ::std::ostream &operator<<(::std::ostream &os, const DataLayout &data_lay case DataLayout::NDHWC: os << "NDHWC"; break; + case DataLayout::NCDHW: + os << "NCDHW"; + break; default: ARM_COMPUTE_ERROR("NOT_SUPPORTED!"); } @@ -707,6 +710,9 @@ inline ::std::ostream &operator<<(::std::ostream &os, const DataLayoutDimension case DataLayoutDimension::CHANNEL: os << "CHANNEL"; break; + case DataLayoutDimension::DEPTH: + os << "DEPTH"; + break; case DataLayoutDimension::BATCHES: os << "BATCHES"; break; -- cgit v1.2.1