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 /src/armnnTfLiteParser/TfLiteParser.hpp | |
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 'src/armnnTfLiteParser/TfLiteParser.hpp')
-rw-r--r-- | src/armnnTfLiteParser/TfLiteParser.hpp | 8 |
1 files changed, 8 insertions, 0 deletions
diff --git a/src/armnnTfLiteParser/TfLiteParser.hpp b/src/armnnTfLiteParser/TfLiteParser.hpp index f8ddc55649..7eb6c48501 100644 --- a/src/armnnTfLiteParser/TfLiteParser.hpp +++ b/src/armnnTfLiteParser/TfLiteParser.hpp @@ -242,7 +242,13 @@ private: }; bool ShouldConstantTensorBeCreated(unsigned int tensorIndex); + bool IsConstTensor(TensorRawPtr tensorPtr); + + bool ShouldConstantTensorBeConverted(TfLiteParserImpl::TensorRawPtr tensorPtr, + armnn::DataType inputDataType, + armnn::DataType filterDataType); + armnn::ConstTensor CreateConstTensorNonPermuted(TensorRawPtr tensorPtr, armnn::TensorInfo& tensorInfo); @@ -250,6 +256,7 @@ private: CreateConstTensorPermuted(TensorRawPtr tensorPtr, armnn::TensorInfo& tensorInfo, armnn::Optional<armnn::PermutationVector&> permutationVector); + std::pair<armnn::ConstTensor, std::unique_ptr<float[]>> CreateConstTensorNonPermuted(TensorRawPtr tensorPtr, armnn::TensorInfo& tensorInfo, @@ -261,6 +268,7 @@ private: TfLiteParserImpl::TensorRawPtr tensorPtr, armnn::TensorInfo& tensorInfo, armnn::Optional<armnn::PermutationVector&> permutationVector); + std::pair<armnn::ConstTensor*, std::unique_ptr<float[]>> CreateConstTensorPtr(TensorRawPtr tensorPtr, armnn::TensorInfo& inputTensorInfo); |