diff options
| author | Mike Kelly <mike.kelly@arm.com> | 2023-01-03 16:29:44 +0000 |
|---|---|---|
| committer | mike.kelly <mike.kelly@arm.com> | 2023-01-05 11:48:13 +0000 |
| commit | 0506ef0a099f5ba564af5e110e6857a68f462080 (patch) | |
| tree | 2ff1a15e435c41916a7f93f14766456759dd20b1 /include | |
| parent | 8b4a483e0e2fee508c23be2248ba0409789f1a74 (diff) | |
| download | armnn-0506ef0a099f5ba564af5e110e6857a68f462080.tar.gz | |
GitHub #543 Problem Parsing Mixed-Precision Model
* Fixed bug when converting Constants with Per-Axis Quantization
Signed-off-by: Mike Kelly <mike.kelly@arm.com>
Change-Id: Ifbea23e60483746ec987da491dae96e74cb33af4
Diffstat (limited to 'include')
| -rw-r--r-- | include/armnnUtils/TensorUtils.hpp | 5 |
1 files changed, 5 insertions, 0 deletions
diff --git a/include/armnnUtils/TensorUtils.hpp b/include/armnnUtils/TensorUtils.hpp index f7f20bd065..2d6ec2fea4 100644 --- a/include/armnnUtils/TensorUtils.hpp +++ b/include/armnnUtils/TensorUtils.hpp @@ -55,4 +55,9 @@ unsigned int GetNumElementsAfter(const armnn::TensorShape& shape, unsigned int a std::pair<unsigned int, std::vector<float>> GetPerAxisParams(const armnn::TensorInfo& info); +template<typename PrimitiveType> +std::unique_ptr<float[]> ToFloatArray(const std::vector<PrimitiveType>& data, const armnn::TensorInfo& tensorInfo); + +std::unique_ptr<float[]> ToFloatArray(const std::vector<uint8_t>& data, const armnn::TensorInfo& tensorInfo); + } // namespace armnnUtils |