From fd0cae3153dd7844b1b14adda8571e4588ed762c Mon Sep 17 00:00:00 2001 From: Sadik Armagan Date: Mon, 8 Nov 2021 17:18:31 +0000 Subject: IVGCVSW-5636 'Implement NNAPI caching functions' * Serialize/Deserialize axis parameter on Softmax Descriptor. Signed-off-by: Sadik Armagan Change-Id: I1bba0dd7d78f9f211f77115dcefccf24a4a61b39 --- src/armnnSerializer/ArmnnSchema.fbs | 1 + 1 file changed, 1 insertion(+) (limited to 'src/armnnSerializer/ArmnnSchema.fbs') diff --git a/src/armnnSerializer/ArmnnSchema.fbs b/src/armnnSerializer/ArmnnSchema.fbs index 40de3496b0..362dd5c653 100644 --- a/src/armnnSerializer/ArmnnSchema.fbs +++ b/src/armnnSerializer/ArmnnSchema.fbs @@ -495,6 +495,7 @@ table SoftmaxLayer { table SoftmaxDescriptor { beta:float; + axis:int = -1; } table DepthwiseConvolution2dLayer { -- cgit v1.2.1