diff options
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) |