diff options
Diffstat (limited to 'src/armnn/StaticRangeVisitor.hpp')
-rw-r--r-- | src/armnn/StaticRangeVisitor.hpp | 7 |
1 files changed, 7 insertions, 0 deletions
diff --git a/src/armnn/StaticRangeVisitor.hpp b/src/armnn/StaticRangeVisitor.hpp index e1f68f3a31..a69df8b705 100644 --- a/src/armnn/StaticRangeVisitor.hpp +++ b/src/armnn/StaticRangeVisitor.hpp @@ -54,6 +54,11 @@ public: const ConstTensor& weights, const Optional<ConstTensor>& biases, const char *name) override; + + void VisitPermuteLayer(const IConnectableLayer* layer, + const PermuteDescriptor& permuteDescriptor, + const char* name) override; + void VisitSoftmaxLayer(const IConnectableLayer* layer, const SoftmaxDescriptor& softmaxDescriptor, const char* name = nullptr) override; @@ -62,6 +67,8 @@ private: /// Set the range for an output slot on a layer void SetRange(const IConnectableLayer* layer, unsigned int outputIdx, float min, float max); + void ForwardParentParameters(const IConnectableLayer* layer); + /// Mapping from a layer Guid to an array of ranges for outputs RangeTracker& m_RangeTracker; }; |