diff options
| author | Nikhil Raj <nikhil.raj@arm.com> | 2019-09-18 17:16:31 +0100 |
|---|---|---|
| committer | Nikhil Raj Arm <nikhil.raj@arm.com> | 2019-09-19 07:34:56 +0000 |
| commit | e1c2d02343f0746950fb5bc2b3cd3a34537a1eee (patch) | |
| tree | 93604eff3eed71cac68aed78e2b9db5b8e12c524 /src/armnn/QuantizerVisitor.hpp | |
| parent | 42f9d9e8fdb15f418240a64a7b089df611a015a0 (diff) | |
| download | armnn-e1c2d02343f0746950fb5bc2b3cd3a34537a1eee.tar.gz | |
IVGCVSW-3725 Adding quantization support for ArgMinMax
Change-Id: I7582a9ee36b4d1764a5a137cefe9b7b7dfe30254
Signed-off-by: Nikhil Raj <nikhil.raj@arm.com>
Diffstat (limited to 'src/armnn/QuantizerVisitor.hpp')
| -rw-r--r-- | src/armnn/QuantizerVisitor.hpp | 4 |
1 files changed, 4 insertions, 0 deletions
diff --git a/src/armnn/QuantizerVisitor.hpp b/src/armnn/QuantizerVisitor.hpp index 3a1e300b7f..db1a669ba8 100644 --- a/src/armnn/QuantizerVisitor.hpp +++ b/src/armnn/QuantizerVisitor.hpp @@ -40,6 +40,10 @@ public: void VisitAdditionLayer(const IConnectableLayer* layer, const char* name = nullptr) override; + void VisitArgMinMaxLayer(const IConnectableLayer* layer, + const ArgMinMaxDescriptor& argMinMaxDescriptor, + const char* name = nullptr) override; + void VisitBatchNormalizationLayer(const IConnectableLayer* layer, const BatchNormalizationDescriptor& desc, const ConstTensor& mean, |