aboutsummaryrefslogtreecommitdiff
path: root/src/armnnSerializer/Serializer.hpp
diff options
context:
space:
mode:
authorRyan OShea <Ryan.OShea2@arm.com>2020-06-05 17:17:06 +0100
committerKeithARM <keith.davis@arm.com>2020-06-11 12:17:49 +0000
commitec6c68093eaef8a2b8e1fd64fcc765237973512e (patch)
treee4dd87466d5be44d1310f9e1c9c6e1355df43fbf /src/armnnSerializer/Serializer.hpp
parent6350d27286114dfdae5f65ae1823ba1150087efb (diff)
downloadarmnn-ec6c68093eaef8a2b8e1fd64fcc765237973512e.tar.gz
IVGCVSW-4906 Add front-end support for FILL operator
* Added new fill layer * Added visitor tests Signed-off-by: Ryan OShea <Ryan.OShea2@arm.com> Change-Id: Iea677014866b4f2d514004623f59ee83f3c0eef8 Signed-off-by: Keith Davis <keith.davis@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 6dd655827d..65d87b7cf7 100644
--- a/src/armnnSerializer/Serializer.hpp
+++ b/src/armnnSerializer/Serializer.hpp
@@ -121,6 +121,10 @@ public:
void VisitEqualLayer(const armnn::IConnectableLayer* layer,
const char* name = nullptr) override;
+ void VisitFillLayer(const armnn::IConnectableLayer* layer,
+ const armnn::FillDescriptor& fillDescriptor,
+ const char* name = nullptr) override;
+
void VisitFloorLayer(const armnn::IConnectableLayer *layer,
const char *name = nullptr) override;