diff options
author | Sadik Armagan <sadik.armagan@arm.com> | 2019-10-14 11:47:07 +0100 |
---|---|---|
committer | Áron Virginás-Tar <aron.virginas-tar@arm.com> | 2019-10-15 09:00:29 +0000 |
commit | 91e1de786ddeee7fcd702a756ef47a475ff1bd46 (patch) | |
tree | a07b26bc9aaa7324f51f388bc870be2cecb119a6 /src/armnn/QuantizerVisitor.hpp | |
parent | 994b5349180b41849ee616fa4f0d3b94a928f48f (diff) | |
download | armnn-91e1de786ddeee7fcd702a756ef47a475ff1bd46.tar.gz |
IVGCVSW-3976 Add Quantizer support for LOG_SOFTMAX
Signed-off-by: Sadik Armagan <sadik.armagan@arm.com>
Change-Id: I907f9b585f7ecd079c9cdd3c2edb6c473bf1d215
Diffstat (limited to 'src/armnn/QuantizerVisitor.hpp')
-rw-r--r-- | src/armnn/QuantizerVisitor.hpp | 4 |
1 files changed, 4 insertions, 0 deletions
diff --git a/src/armnn/QuantizerVisitor.hpp b/src/armnn/QuantizerVisitor.hpp index d1c4375b59..89d1932a08 100644 --- a/src/armnn/QuantizerVisitor.hpp +++ b/src/armnn/QuantizerVisitor.hpp @@ -93,6 +93,10 @@ public: const InstanceNormalizationDescriptor& instanceNormalizationDescriptor, const char* name = nullptr) override; + void VisitLogSoftmaxLayer(const IConnectableLayer* layer, + const LogSoftmaxDescriptor& logSoftmaxDescriptor, + const char* name = nullptr) override; + void VisitMeanLayer(const IConnectableLayer* layer, const MeanDescriptor& meanDescriptor, const char* name = nullptr) override; |