aboutsummaryrefslogtreecommitdiff
path: root/src/armnnSerializer
diff options
context:
space:
mode:
authorColm Donelan <Colm.Donelan@arm.com>2020-02-26 15:39:23 +0000
committerColm Donelan <colm.donelan@arm.com>2020-03-02 21:31:49 +0000
commit03fbeaf532f2575381edc2336f834973117f6e0f (patch)
tree97585ec459e4069853c16468ca82513f0d899200 /src/armnnSerializer
parentc9ea45adefdde2890e9aa191a5b31563a3dd35ea (diff)
downloadarmnn-03fbeaf532f2575381edc2336f834973117f6e0f.tar.gz
IVGCVSW-4440 : Add HARD_SWISH Support to Activation in CpuRef
* Add a new Activiation type of HardSwish. * Add CpuRef support and tests. Signed-off-by: Colm Donelan <Colm.Donelan@arm.com> Change-Id: I68c3840aa45b7a27d5e416a5d50fe8f99f003ce8
Diffstat (limited to 'src/armnnSerializer')
-rw-r--r--src/armnnSerializer/ArmnnSchema.fbs3
-rw-r--r--src/armnnSerializer/Serializer.cpp2
2 files changed, 4 insertions, 1 deletions
diff --git a/src/armnnSerializer/ArmnnSchema.fbs b/src/armnnSerializer/ArmnnSchema.fbs
index d175d41f3f..d7565a5b9a 100644
--- a/src/armnnSerializer/ArmnnSchema.fbs
+++ b/src/armnnSerializer/ArmnnSchema.fbs
@@ -20,7 +20,8 @@ enum ActivationFunction : byte {
Abs = 7,
Sqrt = 8,
Square = 9,
- Elu = 10
+ Elu = 10,
+ HardSwish = 11
}
enum ArgMinMaxFunction : byte {
diff --git a/src/armnnSerializer/Serializer.cpp b/src/armnnSerializer/Serializer.cpp
index a3fdcf8123..39df0c2a7f 100644
--- a/src/armnnSerializer/Serializer.cpp
+++ b/src/armnnSerializer/Serializer.cpp
@@ -48,6 +48,8 @@ serializer::ActivationFunction GetFlatBufferActivationFunction(armnn::Activation
return serializer::ActivationFunction::ActivationFunction_Square;
case armnn::ActivationFunction::Elu:
return serializer::ActivationFunction::ActivationFunction_Elu;
+ case armnn::ActivationFunction::HardSwish:
+ return serializer::ActivationFunction::ActivationFunction_HardSwish;
default:
return serializer::ActivationFunction::ActivationFunction_Sigmoid;
}