diff options
author | Tee Jung <tee.ty.jung@openedges.com> | 2019-11-01 07:04:42 +0000 |
---|---|---|
committer | Matteo Martincigh <matteo.martincigh@arm.com> | 2019-11-04 09:12:46 +0000 |
commit | 7ff9a6096e3c1facbd6786993a6437b9f72069d2 (patch) | |
tree | a4e590d3fbf75d065692b5c52b7bcce6133ed740 /src/armnnOnnxParser/OnnxParser.hpp | |
parent | fcf6fd562f87595c814d8acbec04194421018c32 (diff) | |
download | armnn-7ff9a6096e3c1facbd6786993a6437b9f72069d2.tar.gz |
Make onnx parser to support TanH / Sigmoid / LeakyRelu layers
Signed-off-by: Jung Tae-young tee.ty.jung@openedges.com
Change-Id: I44d24b525b78b8d3fee0197abda7bd667eb04d83
Diffstat (limited to 'src/armnnOnnxParser/OnnxParser.hpp')
-rw-r--r-- | src/armnnOnnxParser/OnnxParser.hpp | 6 |
1 files changed, 6 insertions, 0 deletions
diff --git a/src/armnnOnnxParser/OnnxParser.hpp b/src/armnnOnnxParser/OnnxParser.hpp index 91927c24a8..a467180299 100644 --- a/src/armnnOnnxParser/OnnxParser.hpp +++ b/src/armnnOnnxParser/OnnxParser.hpp @@ -14,6 +14,7 @@ namespace armnn { class TensorInfo; +enum class ActivationFunction; } namespace armnnOnnxParser @@ -103,7 +104,12 @@ private: void AddPoolingLayer(const onnx::NodeProto& nodeProto, armnn::Pooling2dDescriptor& desc); void ParseReshape(const onnx::NodeProto& nodeProto); + + void ParseActivation(const onnx::NodeProto& nodeProto, const armnn::ActivationFunction func); + void ParseSigmoid(const onnx::NodeProto& nodeProto); + void ParseTanh(const onnx::NodeProto& nodeProto); void ParseRelu(const onnx::NodeProto& nodeProto); + void ParseLeakyRelu(const onnx::NodeProto& nodeProto); void AddConvLayerWithDepthwiseConv(const onnx::NodeProto& node, const armnn::Convolution2dDescriptor& convDesc); void ParseConv(const onnx::NodeProto& nodeProto); |