From bf3c6626e98b9e1be435fce9fdabc9d21f3b5b3a Mon Sep 17 00:00:00 2001 From: Michele Di Giorgio Date: Thu, 8 Mar 2018 11:52:27 +0000 Subject: COMPMID-803: Add NHWC data format support for CL batch normalisation Change-Id: Ie37588f60b9cfc7b1d09b1e8628fcfb4b17e0717 Reviewed-on: https://eu-gerrit-1.euhpc.arm.com/123834 Tested-by: Jenkins Reviewed-by: Anthony Barbier --- src/core/Utils.cpp | 12 ++++++++++++ 1 file changed, 12 insertions(+) (limited to 'src/core/Utils.cpp') diff --git a/src/core/Utils.cpp b/src/core/Utils.cpp index 4a237f9daa..b5663e63e4 100644 --- a/src/core/Utils.cpp +++ b/src/core/Utils.cpp @@ -126,6 +126,18 @@ const std::string &arm_compute::string_from_channel(Channel channel) return channels_map[channel]; } +const std::string &arm_compute::string_from_data_layout(DataLayout dl) +{ + static std::map dl_map = + { + { DataLayout::UNKNOWN, "UNKNOWN" }, + { DataLayout::NCHW, "NCHW" }, + { DataLayout::NHWC, "NHWC" }, + }; + + return dl_map[dl]; +} + const std::string &arm_compute::string_from_data_type(DataType dt) { static std::map dt_map = -- cgit v1.2.1