commit 0506ef0a099f5ba564af5e110e6857a68f462080
author Mike Kelly <mike.kelly@arm.com> Tue Jan 03 16:29:44 2023 +0000
committer mike.kelly <mike.kelly@arm.com> Thu Jan 05 11:48:13 2023 +0000
tree 2ff1a15e435c41916a7f93f14766456759dd20b1
parent 8b4a483e0e2fee508c23be2248ba0409789f1a74

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

include/armnnUtils/TensorUtils.hpp

diff --git a/include/armnnUtils/TensorUtils.hpp b/include/armnnUtils/TensorUtils.hpp
index f7f20bd..2d6ec2f 100644
--- a/include/armnnUtils/TensorUtils.hpp
+++ b/include/armnnUtils/TensorUtils.hpp
@@ -55,4 +55,9 @@
 
 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