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.hpp | |
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.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; }; |