diff options
author | Aron Virginas-Tar <Aron.Virginas-Tar@arm.com> | 2019-02-19 16:31:08 +0000 |
---|---|---|
committer | Aron Virginas-Tar <aron.virginas-tar@arm.com> | 2019-02-20 14:08:34 +0000 |
commit | c04125f678be1cce86ebcfe94b587e8eba5b7fde (patch) | |
tree | f6a3616e414f35f7433c04be1cbd15cb9c899d1e /src/armnnSerializer/Schema.fbs | |
parent | 9ffd2fe273c4f98a832da96d37dcb21b94692546 (diff) | |
download | armnn-c04125f678be1cce86ebcfe94b587e8eba5b7fde.tar.gz |
IVGCVSW-2685 Serialize / de-serialize the DepthwiseConvolution2d layer
Change-Id: I37e360c824b30cb14cbef86f6ff7636bc9382109
Signed-off-by: Aron Virginas-Tar <Aron.Virginas-Tar@arm.com>
Diffstat (limited to 'src/armnnSerializer/Schema.fbs')
-rw-r--r-- | src/armnnSerializer/Schema.fbs | 22 |
1 files changed, 21 insertions, 1 deletions
diff --git a/src/armnnSerializer/Schema.fbs b/src/armnnSerializer/Schema.fbs index cbc7da066f..6c542b1b2d 100644 --- a/src/armnnSerializer/Schema.fbs +++ b/src/armnnSerializer/Schema.fbs @@ -75,7 +75,8 @@ enum LayerType : uint { Pooling2d = 4, Reshape = 5, Softmax = 6, - Convolution2d = 7 + Convolution2d = 7, + DepthwiseConvolution2d = 8 } // Base layer table to be used as part of other layers @@ -168,6 +169,24 @@ table SoftmaxDescriptor { beta:float; } +table DepthwiseConvolution2dLayer { + base:LayerBase; + descriptor:DepthwiseConvolution2dDescriptor; + weights:ConstTensor; + biases:ConstTensor; +} + +table DepthwiseConvolution2dDescriptor { + padLeft:uint; + padRight:uint; + padTop:uint; + padBottom:uint; + strideX:uint; + strideY:uint; + biasEnabled:bool = false; + dataLayout:DataLayout = NCHW; +} + table OutputLayer { base:BindableLayerBase; } @@ -184,6 +203,7 @@ table ReshapeDescriptor { union Layer { AdditionLayer, Convolution2dLayer, + DepthwiseConvolution2dLayer, InputLayer, MultiplicationLayer, OutputLayer, |