diff options
author | Ellen Norris-Thompson <ellen.norris-thompson@arm.com> | 2019-06-19 11:46:21 +0100 |
---|---|---|
committer | Matteo Martincigh <matteo.martincigh@arm.com> | 2019-06-19 16:27:26 +0000 |
commit | 51982472bfedf12e7d82cde6614617f94b2c86d0 (patch) | |
tree | 0e2bef4812e0a3128b47a81f393dd6583d45e4c7 /src/armnnSerializer/ArmnnSchema.fbs | |
parent | a4812b6cd54e4dc4903f457066281d8bf0ccf448 (diff) | |
download | armnn-51982472bfedf12e7d82cde6614617f94b2c86d0.tar.gz |
IVGCVSW-3269 Add Serialization support for the new Prelu Activation layer
* Adds serialization/deserialization support
* Adds related unit test
Signed-off-by: Ellen Norris-Thompson <ellen.norris-thompson@arm.com>
Change-Id: I600322b03e51f443cbcd9262bb27e36e5fd95ae5
Diffstat (limited to 'src/armnnSerializer/ArmnnSchema.fbs')
-rw-r--r-- | src/armnnSerializer/ArmnnSchema.fbs | 10 |
1 files changed, 8 insertions, 2 deletions
diff --git a/src/armnnSerializer/ArmnnSchema.fbs b/src/armnnSerializer/ArmnnSchema.fbs index 1204ad8f0f..db5672f948 100644 --- a/src/armnnSerializer/ArmnnSchema.fbs +++ b/src/armnnSerializer/ArmnnSchema.fbs @@ -122,7 +122,8 @@ enum LayerType : uint { Merge = 37, Switch = 38, Concat = 39, - SpaceToDepth = 40 + SpaceToDepth = 40, + Prelu = 41 } // Base layer table to be used as part of other layers @@ -555,6 +556,10 @@ table SwitchLayer { base:LayerBase; } +table PreluLayer { + base:LayerBase; +} + union Layer { ActivationLayer, AdditionLayer, @@ -596,7 +601,8 @@ union Layer { MergeLayer, SwitchLayer, ConcatLayer, - SpaceToDepthLayer + SpaceToDepthLayer, + PreluLayer } table AnyLayer { |