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.hpp | |
| 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.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 1beaf5adde..ac90bfb1e5 100644 --- a/src/armnn/QuantizerVisitor.hpp +++ b/src/armnn/QuantizerVisitor.hpp @@ -84,6 +84,10 @@ public: const ConstTensor& input, const char* name = nullptr) override; + void VisitMergerLayer(const IConnectableLayer* layer, + const OriginsDescriptor& mergerDescriptor, + const char* name = nullptr) override; + /// Extract the quantized network INetworkPtr RetrieveFinalNetwork() { return std::move(m_QuantizedNetwork); } |