aboutsummaryrefslogtreecommitdiff
path: root/src/armnnSerializer/Serializer.hpp
diff options
context:
space:
mode:
authorJim Flynn <jim.flynn@arm.com>2019-03-08 11:08:30 +0000
committerJim Flynn <jim.flynn@arm.com>2019-03-12 10:43:34 +0000
commit18ce338711fc3ea44a7731eac795964256beac6c (patch)
tree93467a86f34cea694e08bf539b3d315680771a3a /src/armnnSerializer/Serializer.hpp
parent1b4d7b1e0646104210f626e7f9613c506f097e90 (diff)
downloadarmnn-18ce338711fc3ea44a7731eac795964256beac6c.tar.gz
IVGCVSW-2709 Serialize / de-serialize the Splitter layer
* fixed typo in Ref Merger Workload comment * fixed typo in ViewsDescriptor comment * made the origins descriptor accessable in the ViewsDescriptor (needed for serialization) * based the unit test on the use of the splitter in the CaffeParser Change-Id: I3e716839adb4eee5a695633377b49e7e18ec2aa9 Signed-off-by: Ferran Balaguer <ferran.balaguer@arm.com> Signed-off-by: Francis Murtagh <francis.murtagh@arm.com> Signed-off-by: Jim Flynn <jim.flynn@arm.com>
Diffstat (limited to 'src/armnnSerializer/Serializer.hpp')
-rw-r--r--src/armnnSerializer/Serializer.hpp4
1 files changed, 4 insertions, 0 deletions
diff --git a/src/armnnSerializer/Serializer.hpp b/src/armnnSerializer/Serializer.hpp
index c0e70c982c..cb05792863 100644
--- a/src/armnnSerializer/Serializer.hpp
+++ b/src/armnnSerializer/Serializer.hpp
@@ -162,6 +162,10 @@ public:
const armnn::NormalizationDescriptor& normalizationDescriptor,
const char* name = nullptr) override;
+ void VisitSplitterLayer(const armnn::IConnectableLayer* layer,
+ const armnn::ViewsDescriptor& viewsDescriptor,
+ const char* name = nullptr) override;
+
void VisitStridedSliceLayer(const armnn::IConnectableLayer* layer,
const armnn::StridedSliceDescriptor& stridedSliceDescriptor,
const char* name = nullptr) override;