From 16dddd2af57a71ca10d62a4412d014f859720d2c Mon Sep 17 00:00:00 2001 From: Sheri Zhang Date: Wed, 27 May 2020 15:03:48 +0100 Subject: COMPMID-3381: Implement graph example for YoLo v3 output detector Add sub/exp/splitv support in graph api Signed-off-by: Sheri Zhang Change-Id: I4e08cc19a46655717068b12c93d67e619a595d9a Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/3309 Reviewed-by: Georgios Pinitas Tested-by: Arm Jenkins Comments-Addressed: Arm Jenkins --- arm_compute/graph/Types.h | 9 ++++++++- 1 file changed, 8 insertions(+), 1 deletion(-) (limited to 'arm_compute/graph/Types.h') diff --git a/arm_compute/graph/Types.h b/arm_compute/graph/Types.h index 296f757c9b..422069238d 100644 --- a/arm_compute/graph/Types.h +++ b/arm_compute/graph/Types.h @@ -103,7 +103,13 @@ enum class EltwiseOperation { Add, /**< Arithmetic addition */ Sub, /**< Arithmetic subtraction */ - Mul /**< Arithmetic multiplication */ + Mul, /**< Arithmetic multiplication */ +}; + +/** Supported Unary Element-wise operations */ +enum class UnaryEltwiseOperation +{ + Exp /**< Exp */ }; /** Supported Convolution layer methods */ @@ -168,6 +174,7 @@ enum class NodeType SplitLayer, StackLayer, UpsampleLayer, + UnaryEltwiseLayer, YOLOLayer, Input, -- cgit v1.2.1