diff options
Diffstat (limited to 'src/armnnSerializer/Schema.fbs')
-rw-r--r-- | src/armnnSerializer/Schema.fbs | 15 |
1 files changed, 13 insertions, 2 deletions
diff --git a/src/armnnSerializer/Schema.fbs b/src/armnnSerializer/Schema.fbs index 5d6388d944..411b89ad8e 100644 --- a/src/armnnSerializer/Schema.fbs +++ b/src/armnnSerializer/Schema.fbs @@ -66,7 +66,8 @@ enum LayerType : uint { Addition = 0, Input = 1, Multiplication = 2, - Output = 3 + Output = 3, + Softmax = 4 } // Base layer table to be used as part of other layers @@ -96,6 +97,15 @@ table MultiplicationLayer { base:LayerBase; } +table SoftmaxLayer { + base:LayerBase; + descriptor:SoftmaxDescriptor; +} + +table SoftmaxDescriptor { + beta:float; +} + table OutputLayer { base:BindableLayerBase; } @@ -104,7 +114,8 @@ union Layer { AdditionLayer, InputLayer, MultiplicationLayer, - OutputLayer + OutputLayer, + SoftmaxLayer } table AnyLayer { |