aboutsummaryrefslogtreecommitdiff
path: root/src/armnn/SerializeLayerParameters.cpp
diff options
context:
space:
mode:
authorTeresa Charlin <teresa.charlinreyes@arm.com>2023-07-27 14:11:54 +0100
committerTeresaARM <teresa.charlinreyes@arm.com>2023-07-28 16:59:37 +0000
commita0cc6ff8065083b833dda9c3c5992938834bdf29 (patch)
treea5bacca27431c429ac982c44b0a698e6b20bfe03 /src/armnn/SerializeLayerParameters.cpp
parent65f9504e7381a28cda5f02c0e1bca65a1e2b90b2 (diff)
downloadarmnn-a0cc6ff8065083b833dda9c3c5992938834bdf29.tar.gz
IVGCVSW-7924 Add TILE to Support Library (SL)
* Add serialize parameters so that the multiples appear in the dot file Signed-off-by: Teresa Charlin <teresa.charlinreyes@arm.com> Change-Id: Id02ed709c97b866dabefed655b06bdb1b20c9026
Diffstat (limited to 'src/armnn/SerializeLayerParameters.cpp')
-rw-r--r--src/armnn/SerializeLayerParameters.cpp17
1 files changed, 17 insertions, 0 deletions
diff --git a/src/armnn/SerializeLayerParameters.cpp b/src/armnn/SerializeLayerParameters.cpp
index 67520a72d6..1445c70a70 100644
--- a/src/armnn/SerializeLayerParameters.cpp
+++ b/src/armnn/SerializeLayerParameters.cpp
@@ -558,6 +558,23 @@ void StringifyLayerParameters<StridedSliceDescriptor>::Serialize(ParameterString
fn("DataLayout", GetDataLayoutName(desc.m_DataLayout));
}
+void StringifyLayerParameters<TileDescriptor>::Serialize(ParameterStringifyFunction& fn,
+ const TileDescriptor& desc)
+{
+ std::stringstream ss;
+ int count = 0;
+ for (auto&& var : desc.m_Multiples)
+ {
+ if (count > 0)
+ {
+ ss << ",";
+ }
+ ss << var;
+ ++count;
+ }
+ fn("Multiples", ss.str());
+}
+
void StringifyLayerParameters<TransposeConvolution2dDescriptor>::Serialize(
ParameterStringifyFunction& fn,
const TransposeConvolution2dDescriptor& desc)