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/StaticRangeVisitor.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/StaticRangeVisitor.cpp')
-rw-r--r-- | src/armnn/StaticRangeVisitor.cpp | 14 |
1 files changed, 14 insertions, 0 deletions
diff --git a/src/armnn/StaticRangeVisitor.cpp b/src/armnn/StaticRangeVisitor.cpp index 2365e1b38c..ecdc1ad957 100644 --- a/src/armnn/StaticRangeVisitor.cpp +++ b/src/armnn/StaticRangeVisitor.cpp @@ -21,6 +21,12 @@ void StaticRangeVisitor::SetRange(const IConnectableLayer* layer, unsigned int o m_RangeTracker.SetRange(layer, outputIdx, min, max); } +void StaticRangeVisitor::ForwardParentParameters(const IConnectableLayer* layer) +{ + const auto parentRange = m_RangeTracker.GetRange(layer->GetInputSlot(0).GetConnection()->GetOwningLayerGuid(), 0); + SetRange(layer, 0, parentRange.first, parentRange.second); +} + void StaticRangeVisitor::VisitAdditionLayer(const IConnectableLayer* layer, const char* name) { SetRange(layer, 0, -20.f, 20.f); @@ -110,6 +116,14 @@ void StaticRangeVisitor::VisitFullyConnectedLayer(const IConnectableLayer *layer SetRange(layer, 0, -15.0f, 15.0f); } +void StaticRangeVisitor::VisitPermuteLayer(const IConnectableLayer* layer, + const PermuteDescriptor& permuteDescriptor, + const char* name) +{ + boost::ignore_unused(permuteDescriptor); + ForwardParentParameters(layer); +} + void StaticRangeVisitor::VisitSoftmaxLayer(const IConnectableLayer* layer, const SoftmaxDescriptor& softmaxDescriptor, const char* name) |