diff options
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); |