diff options
| author | keidav01 <keith.davis@arm.com> | 2019-02-21 10:07:37 +0000 |
|---|---|---|
| committer | Aron Virginas-Tar <aron.virginas-tar@arm.com> | 2019-02-21 11:08:38 +0000 |
| commit | 1b3e2ead82db933ea8e97063cea132cb042b079a (patch) | |
| tree | ed6df4aafb7a813cbf1c77139b24536ada3ce84e /include | |
| parent | 19f74b440ab61a8f39aefe68bd943f7fafc4adc6 (diff) | |
| download | armnn-1b3e2ead82db933ea8e97063cea132cb042b079a.tar.gz | |
IVGCVSW-2429 Add Detection PostProcess Parser to TensorFlow Lite Parser
* Added parser function in TFLiteParser
* Removed custom options gating
* Added unit test
* Removed template usage in VerifyTensorInfo for DeserializeParser
Change-Id: If198654ed70060855a05f8aaed010293405bd103
Signed-off-by: keidav01 <keith.davis@arm.com>
Diffstat (limited to 'include')
| -rw-r--r-- | include/armnn/TypesUtils.hpp | 9 |
1 files changed, 4 insertions, 5 deletions
diff --git a/include/armnn/TypesUtils.hpp b/include/armnn/TypesUtils.hpp index bb75b18c32..c65eefc510 100644 --- a/include/armnn/TypesUtils.hpp +++ b/include/armnn/TypesUtils.hpp @@ -219,15 +219,14 @@ inline float Dequantize(QuantizedType value, float scale, int32_t offset) return dequantized; } -template <armnn::DataType DataType> -void VerifyTensorInfoDataType(const armnn::TensorInfo & info) +inline void VerifyTensorInfoDataType(const armnn::TensorInfo & info, armnn::DataType dataType) { - if (info.GetDataType() != DataType) + if (info.GetDataType() != dataType) { std::stringstream ss; ss << "Unexpected datatype:" << armnn::GetDataTypeName(info.GetDataType()) - << " for tensor:" << info.GetShape() - << ". The type expected to be: " << armnn::GetDataTypeName(DataType); + << " for tensor:" << info.GetShape() + << ". The type expected to be: " << armnn::GetDataTypeName(dataType); throw armnn::Exception(ss.str()); } } |