diff options
author | FrancisMurtagh <francis.murtagh@arm.com> | 2019-02-12 16:52:24 +0000 |
---|---|---|
committer | Francis Murtagh <francis.murtagh@arm.com> | 2019-02-12 16:58:26 +0000 |
commit | fb1437e86d8e01af9ee9cebe4c8cd9ff508ac779 (patch) | |
tree | 609fa13934dd10944efa6c43dc1f2966e82ba309 /src/armnn/QuantizerVisitor.cpp | |
parent | 8a4ca10cca8d5dadb52124187c209c955df7a8b6 (diff) | |
download | armnn-fb1437e86d8e01af9ee9cebe4c8cd9ff508ac779.tar.gz |
IVGCVSW-2614 Support static quantization of Permute
Change-Id: I4f208d18f7ac6bb04423cc26ecf61adf0ccc5d26
Signed-off-by: FrancisMurtagh <francis.murtagh@arm.com>
Diffstat (limited to 'src/armnn/QuantizerVisitor.cpp')
-rw-r--r-- | src/armnn/QuantizerVisitor.cpp | 10 |
1 files changed, 10 insertions, 0 deletions
diff --git a/src/armnn/QuantizerVisitor.cpp b/src/armnn/QuantizerVisitor.cpp index af01092f8e..df5dd59677 100644 --- a/src/armnn/QuantizerVisitor.cpp +++ b/src/armnn/QuantizerVisitor.cpp @@ -205,6 +205,16 @@ void QuantizerVisitor::VisitDepthwiseConvolution2dLayer(const IConnectableLayer* SetQuantizedInputConnections(layer, newLayer); } + +void QuantizerVisitor::VisitPermuteLayer(const IConnectableLayer* layer, + const PermuteDescriptor& permuteDescriptor, + const char* name) +{ + IConnectableLayer* newLayer = m_QuantizedNetwork->AddPermuteLayer(permuteDescriptor, name); + RecordLayer(layer, newLayer); + SetQuantizedInputConnections(layer, newLayer); +} + void QuantizerVisitor::VisitSoftmaxLayer(const IConnectableLayer* layer, const SoftmaxDescriptor& softmaxDescriptor, const char* name) |