From 91e1de786ddeee7fcd702a756ef47a475ff1bd46 Mon Sep 17 00:00:00 2001 From: Sadik Armagan Date: Mon, 14 Oct 2019 11:47:07 +0100 Subject: IVGCVSW-3976 Add Quantizer support for LOG_SOFTMAX Signed-off-by: Sadik Armagan Change-Id: I907f9b585f7ecd079c9cdd3c2edb6c473bf1d215 --- src/armnn/QuantizerVisitor.cpp | 13 +++++++++++-- 1 file changed, 11 insertions(+), 2 deletions(-) (limited to 'src/armnn/QuantizerVisitor.cpp') diff --git a/src/armnn/QuantizerVisitor.cpp b/src/armnn/QuantizerVisitor.cpp index 9819d71ea9..7158c99995 100644 --- a/src/armnn/QuantizerVisitor.cpp +++ b/src/armnn/QuantizerVisitor.cpp @@ -319,9 +319,18 @@ void QuantizerVisitor::VisitInstanceNormalizationLayer(const IConnectableLayer* SetQuantizedInputConnections(layer, newLayer); } +void QuantizerVisitor::VisitLogSoftmaxLayer(const IConnectableLayer* layer, + const LogSoftmaxDescriptor& logSoftmaxDescriptor, + const char* name) +{ + IConnectableLayer* newLayer = m_QuantizedNetwork->AddLogSoftmaxLayer(logSoftmaxDescriptor, name); + RecordLayer(layer, newLayer); + SetQuantizedInputConnections(layer, newLayer); +} + void QuantizerVisitor::VisitMeanLayer(const IConnectableLayer* layer, - const MeanDescriptor& meanDescriptor, - const char* name) + const MeanDescriptor& meanDescriptor, + const char* name) { IConnectableLayer* newLayer = m_QuantizedNetwork->AddMeanLayer(meanDescriptor, name); RecordLayer(layer, newLayer); -- cgit v1.2.1