diff options
| author | Jim Flynn <jim.flynn@arm.com> | 2019-02-14 09:56:36 +0000 |
|---|---|---|
| committer | Matteo Martincigh <matteo.martincigh@arm.com> | 2019-02-15 15:00:11 +0000 |
| commit | e870bef8c93a3c711362b3530aa49c353f048d01 (patch) | |
| tree | e2c63e9ef64b345c837ecfd656ecd42f14ac2221 /src/armnn/QuantizerVisitor.cpp | |
| parent | 43e7864cf81cd55fb53f8b98b4d387da859eae0d (diff) | |
| download | armnn-e870bef8c93a3c711362b3530aa49c353f048d01.tar.gz | |
IVGCVSW-2621 Add static quantization of Merger
Change-Id: I19f01698a6f9b361cd1737d76e6ec2020fab77a6
Signed-off-by: Jim Flynn <jim.flynn@arm.com>
Diffstat (limited to 'src/armnn/QuantizerVisitor.cpp')
| -rw-r--r-- | src/armnn/QuantizerVisitor.cpp | 9 |
1 files changed, 9 insertions, 0 deletions
diff --git a/src/armnn/QuantizerVisitor.cpp b/src/armnn/QuantizerVisitor.cpp index 8e3265fb75..88062068e8 100644 --- a/src/armnn/QuantizerVisitor.cpp +++ b/src/armnn/QuantizerVisitor.cpp @@ -233,6 +233,15 @@ void QuantizerVisitor::VisitPooling2dLayer(const IConnectableLayer* layer, SetQuantizedInputConnections(layer, newLayer); } +void QuantizerVisitor::VisitMergerLayer(const IConnectableLayer* layer, + const OriginsDescriptor& mergerDescriptor, + const char* name) +{ + IConnectableLayer* newLayer = m_QuantizedNetwork->AddMergerLayer(mergerDescriptor, name); + RecordLayer(layer, newLayer); + SetQuantizedInputConnections(layer, newLayer); +} + void QuantizerVisitor::VisitSoftmaxLayer(const IConnectableLayer* layer, const SoftmaxDescriptor& softmaxDescriptor, const char* name) |