diff options
author | Mike Kelly <mike.kelly@arm.com> | 2023-08-08 12:00:28 +0100 |
---|---|---|
committer | Mike Kelly <mike.kelly@arm.com> | 2023-08-08 13:16:17 +0100 |
commit | fca5916e4e6a44cf11b47328659d4d7ee95ec231 (patch) | |
tree | 0810bdfde4137dbee99304f1d6fa2033db084af8 /src/armnn/SerializeLayerParameters.cpp | |
parent | 3b3dcbf0321fadcb2b7b5b550a4d03f510d7cb7b (diff) | |
download | armnn-fca5916e4e6a44cf11b47328659d4d7ee95ec231.tar.gz |
MLCE-1093 Added Axis to ViewsDescriptor
* Added Axis to ViewsDescriptor to store the value where ever possible.
* Updated Serializer and Deserializer to handle axis.
Signed-off-by: Mike Kelly <mike.kelly@arm.com>
Change-Id: I56e442872b47485a608b25fbc79063b362a25618
Diffstat (limited to 'src/armnn/SerializeLayerParameters.cpp')
-rw-r--r-- | src/armnn/SerializeLayerParameters.cpp | 4 |
1 files changed, 4 insertions, 0 deletions
diff --git a/src/armnn/SerializeLayerParameters.cpp b/src/armnn/SerializeLayerParameters.cpp index 1445c70a70..d65a7d55fa 100644 --- a/src/armnn/SerializeLayerParameters.cpp +++ b/src/armnn/SerializeLayerParameters.cpp @@ -636,6 +636,10 @@ void StringifyLayerParameters<ViewsDescriptor>::Serialize(ParameterStringifyFunc } value << "]"; fn(key.str(), value.str()); + if (desc.HasAxis()) + { + fn("Axis", std::to_string(desc.GetAxis())); + } } StringifyLayerParameters<OriginsDescriptor>::Serialize(fn, desc.GetOrigins()); } |