diff options
author | Sadik Armagan <sadik.armagan@arm.com> | 2019-03-04 17:44:21 +0000 |
---|---|---|
committer | Sadik Armagan <sadik.armagan@arm.com> | 2019-03-07 09:24:37 +0000 |
commit | ac97c8cda28f81ce76834b8b769967d42b02e2ac (patch) | |
tree | 7bf1b1104a4bfb041ee5d74cabe1e1be2664af70 /src/armnnSerializer/ArmnnSchema.fbs | |
parent | c192f35edefd0977396db8d381adc7598e3660cc (diff) | |
download | armnn-ac97c8cda28f81ce76834b8b769967d42b02e2ac.tar.gz |
IVGCVSW-2696 Serialize / de-serialize the Mean layer
Change-Id: Iee4bab5a6d6b992cf4bba8697a2918f854c906a3
Signed-off-by: Sadik Armagan <sadik.armagan@arm.com>
Diffstat (limited to 'src/armnnSerializer/ArmnnSchema.fbs')
-rw-r--r-- | src/armnnSerializer/ArmnnSchema.fbs | 16 |
1 files changed, 14 insertions, 2 deletions
diff --git a/src/armnnSerializer/ArmnnSchema.fbs b/src/armnnSerializer/ArmnnSchema.fbs index ac32e661c0..01142ff391 100644 --- a/src/armnnSerializer/ArmnnSchema.fbs +++ b/src/armnnSerializer/ArmnnSchema.fbs @@ -109,7 +109,8 @@ enum LayerType : uint { ResizeBilinear = 25, Subtraction = 26, StridedSlice = 27, - Gather = 28 + Gather = 28, + Mean = 29 } // Base layer table to be used as part of other layers @@ -349,6 +350,16 @@ table NormalizationDescriptor { dataLayout:DataLayout = NCHW; } +table MeanLayer { + base:LayerBase; + descriptor:MeanDescriptor; +} + +table MeanDescriptor { + axis:[uint]; + keepDims:bool = false; +} + table PadLayer { base:LayerBase; descriptor:PadDescriptor; @@ -433,7 +444,8 @@ union Layer { ResizeBilinearLayer, SubtractionLayer, StridedSliceLayer, - GatherLayer + GatherLayer, + MeanLayer } table AnyLayer { |