aboutsummaryrefslogtreecommitdiff
path: root/src/armnnSerializer/Serializer.hpp
diff options
context:
space:
mode:
authorFinn Williams <finn.williams@arm.com>2019-03-01 11:51:52 +0000
committerFinn Williams <finn.williams@arm.com>2019-03-01 15:22:21 +0000
commitdd2ba7ebf78a75aadd8ddd2ae1a4226ffc4ae4d9 (patch)
tree9e0b00f2c80f57f3787087bf7377bae0b0a90223 /src/armnnSerializer/Serializer.hpp
parent8b42a380b5cd4dae367c8bf9a2566985ed92c516 (diff)
downloadarmnn-dd2ba7ebf78a75aadd8ddd2ae1a4226ffc4ae4d9.tar.gz
IVGCVSW-2689 Serialize / de-serialize the Floor layer
Change-Id: I61177a999092195fbdfdcf810d289969252dd365 Signed-off-by: Finn Williams <finn.williams@arm.com>
Diffstat (limited to 'src/armnnSerializer/Serializer.hpp')
-rw-r--r--src/armnnSerializer/Serializer.hpp3
1 files changed, 3 insertions, 0 deletions
diff --git a/src/armnnSerializer/Serializer.hpp b/src/armnnSerializer/Serializer.hpp
index 6e92a067b6..bf599b1a1f 100644
--- a/src/armnnSerializer/Serializer.hpp
+++ b/src/armnnSerializer/Serializer.hpp
@@ -75,6 +75,9 @@ public:
void VisitEqualLayer(const armnn::IConnectableLayer* layer,
const char* name = nullptr) override;
+ void VisitFloorLayer(const armnn::IConnectableLayer *layer,
+ const char *name = nullptr) override;
+
void VisitFullyConnectedLayer(const armnn::IConnectableLayer* layer,
const armnn::FullyConnectedDescriptor& fullyConnectedDescriptor,
const armnn::ConstTensor& weights,