diff options
author | Teresa Charlin <teresa.charlinreyes@arm.com> | 2023-12-15 14:20:47 +0000 |
---|---|---|
committer | john.mcloughlin <john.mcloughlin@arm.com> | 2023-12-21 11:14:54 +0000 |
commit | ca5a23a7cbe46b8da8de432d80889c47a745ca4c (patch) | |
tree | c375efe1f4ecdad708d6e8e771a13c07ddbc0257 /src/backends/tosaCommon/TosaMappings.cpp | |
parent | 0587dd01272199a36210bd0ccc266185b113df75 (diff) | |
download | armnn-ca5a23a7cbe46b8da8de432d80889c47a745ca4c.tar.gz |
Add Quantize Support to TOSA Ref Backend
* Adding a one to many tosa mapping for Quantize
* Added tests
* Resolves IVGCVSW-7175
Signed-off-by: John Mcloughlin <john.mcloughlin@arm.com>
Signed-off-by: Teresa Charlin <teresa.charlinreyes@arm.com>
Change-Id: Ia0852fefb618b4a29c2601b9de8b6b2731229801
Diffstat (limited to 'src/backends/tosaCommon/TosaMappings.cpp')
-rw-r--r-- | src/backends/tosaCommon/TosaMappings.cpp | 4 |
1 files changed, 4 insertions, 0 deletions
diff --git a/src/backends/tosaCommon/TosaMappings.cpp b/src/backends/tosaCommon/TosaMappings.cpp index dff266d793..6c6bff4087 100644 --- a/src/backends/tosaCommon/TosaMappings.cpp +++ b/src/backends/tosaCommon/TosaMappings.cpp @@ -74,6 +74,10 @@ TosaSerializationBasicBlock* GetTosaMapping(const Layer* layer, return ConvertPooling2DToTosaOperator(layer, inputs, outputs, poolDesc); } } + case LayerType::Quantize: + { + return ConvertQuantizeToTosaOperator(layer, inputs, outputs); + } case LayerType::Reshape: { auto reshapeDesc = PolymorphicDowncast<const ReshapeDescriptor*>(&descriptor); |