diff options
author | Giorgio Arena <giorgio.arena@arm.com> | 2021-10-06 12:54:29 +0100 |
---|---|---|
committer | Giorgio Arena <giorgio.arena@arm.com> | 2021-10-07 11:34:26 +0000 |
commit | c9fe9fc3fc8c58a9d724776f75831ab35f07b253 (patch) | |
tree | 3d0ca5dfa67fc62bd063925dc0bc179ad0d08c5a /utils/TypePrinter.h | |
parent | b6af482bc5d8e4f03f876e17909c561de198c4d3 (diff) | |
download | ComputeLibrary-c9fe9fc3fc8c58a9d724776f75831ab35f07b253.tar.gz |
Add support for 5D data layout indexing
Signed-off-by: Giorgio Arena <giorgio.arena@arm.com>
Change-Id: Ib346bb6b90d2220ec5934c83a9a1f0cd540b8731
Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/6377
Comments-Addressed: Arm Jenkins <bsgcomp@arm.com>
Reviewed-by: Gunes Bayir <gunes.bayir@arm.com>
Tested-by: Arm Jenkins <bsgcomp@arm.com>
Diffstat (limited to 'utils/TypePrinter.h')
-rw-r--r-- | utils/TypePrinter.h | 6 |
1 files changed, 6 insertions, 0 deletions
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; |