diff options
Diffstat (limited to 'arm_compute/graph/Types.h')
-rw-r--r-- | arm_compute/graph/Types.h | 9 |
1 files changed, 8 insertions, 1 deletions
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, |