From c9ea45adefdde2890e9aa191a5b31563a3dd35ea Mon Sep 17 00:00:00 2001 From: Mike Kelly Date: Fri, 28 Feb 2020 18:11:58 +0000 Subject: IVGCVSW-4375 Add support for Transpose * Added TransposeLayer * Added CL, Neon and Ref Workloads * Added Transpose utilities * Added Serializer and Deserializer support * Added Quantizer support Signed-off-by: Mike Kelly Change-Id: I04c755ba7cb5b1edf72b3c9f3c0314878032e3c7 --- src/armnn/SerializeLayerParameters.cpp | 20 ++++++++++++++++++++ 1 file changed, 20 insertions(+) (limited to 'src/armnn/SerializeLayerParameters.cpp') diff --git a/src/armnn/SerializeLayerParameters.cpp b/src/armnn/SerializeLayerParameters.cpp index 544e389ae9..76b92f3f9d 100644 --- a/src/armnn/SerializeLayerParameters.cpp +++ b/src/armnn/SerializeLayerParameters.cpp @@ -491,4 +491,24 @@ void StringifyLayerParameters::Serialize( fn("DataLayout", GetDataLayoutName(desc.m_DataLayout)); } +void StringifyLayerParameters::Serialize(ParameterStringifyFunction& fn, + const TransposeDescriptor& desc) +{ + std::stringstream ss; + ss << "["; + bool addComma = false; + for (auto it : desc.m_DimMappings) + { + if (addComma) + { + ss << ","; + } + ss << it; + addComma = true; + } + ss << "]"; + + fn("DimMappings",ss.str()); +} + } // namespace armnn \ No newline at end of file -- cgit v1.2.1