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 /src/armnnDeserializer | |
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 'src/armnnDeserializer')
-rw-r--r-- | src/armnnDeserializer/test/ParserFlatbuffersSerializeFixture.hpp | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/src/armnnDeserializer/test/ParserFlatbuffersSerializeFixture.hpp b/src/armnnDeserializer/test/ParserFlatbuffersSerializeFixture.hpp index 42ab2b17d6..e5416362b6 100644 --- a/src/armnnDeserializer/test/ParserFlatbuffersSerializeFixture.hpp +++ b/src/armnnDeserializer/test/ParserFlatbuffersSerializeFixture.hpp @@ -171,7 +171,7 @@ void ParserFlatbuffersSerializeFixture::RunTest(unsigned int layersId, for (auto&& it : inputData) { BindingPointInfo bindingInfo = m_Parser->GetNetworkInputBindingInfo(layersId, it.first); - armnn::VerifyTensorInfoDataType<ArmnnType>(bindingInfo.second); + armnn::VerifyTensorInfoDataType(bindingInfo.second, ArmnnType); inputTensors.push_back({ bindingInfo.first, armnn::ConstTensor(bindingInfo.second, it.second.data()) }); } @@ -181,7 +181,7 @@ void ParserFlatbuffersSerializeFixture::RunTest(unsigned int layersId, for (auto&& it : expectedOutputData) { BindingPointInfo bindingInfo = m_Parser->GetNetworkOutputBindingInfo(layersId, it.first); - armnn::VerifyTensorInfoDataType<ArmnnType>(bindingInfo.second); + armnn::VerifyTensorInfoDataType(bindingInfo.second, ArmnnType); outputStorage.emplace(it.first, MakeTensor<DataType, NumOutputDimensions>(bindingInfo.second)); outputTensors.push_back( { bindingInfo.first, armnn::Tensor(bindingInfo.second, outputStorage.at(it.first).data()) }); @@ -196,4 +196,4 @@ void ParserFlatbuffersSerializeFixture::RunTest(unsigned int layersId, auto outputExpected = MakeTensor<DataType, NumOutputDimensions>(bindingInfo.second, it.second); BOOST_TEST(CompareTensors(outputExpected, outputStorage[it.first])); } -} +}
\ No newline at end of file |