From 762778817f6567f405d5d705f9c2131bab799e66 Mon Sep 17 00:00:00 2001 From: Conor Kennedy Date: Tue, 26 Feb 2019 08:29:54 +0000 Subject: IVGCVSW-2683 Add Serializer & Deserializer for Constant Change-Id: Iad7d89dfa963d9015cbe044f67aecc8bf6634b10 Signed-off-by: Conor Kennedy --- src/armnnSerializer/ArmnnSchema.fbs | 9 ++++++++- 1 file changed, 8 insertions(+), 1 deletion(-) (limited to 'src/armnnSerializer/ArmnnSchema.fbs') diff --git a/src/armnnSerializer/ArmnnSchema.fbs b/src/armnnSerializer/ArmnnSchema.fbs index dc14069798..db70e7bd4c 100644 --- a/src/armnnSerializer/ArmnnSchema.fbs +++ b/src/armnnSerializer/ArmnnSchema.fbs @@ -92,7 +92,8 @@ enum LayerType : uint { DepthwiseConvolution2d = 8, Activation = 9, Permute = 10, - FullyConnected = 11 + FullyConnected = 11, + Constant = 12 } // Base layer table to be used as part of other layers @@ -125,6 +126,11 @@ table AdditionLayer { base:LayerBase; } +table ConstantLayer { + base:LayerBase; + input:ConstTensor; +} + table Convolution2dLayer { base:LayerBase; descriptor:Convolution2dDescriptor; @@ -251,6 +257,7 @@ table PermuteDescriptor { union Layer { ActivationLayer, AdditionLayer, + ConstantLayer, Convolution2dLayer, DepthwiseConvolution2dLayer, FullyConnectedLayer, -- cgit v1.2.1