diff options
Diffstat (limited to 'src/armnnTfLiteParser/test')
-rw-r--r-- | src/armnnTfLiteParser/test/ParserFlatbuffersFixture.hpp | 6 |
1 files changed, 4 insertions, 2 deletions
diff --git a/src/armnnTfLiteParser/test/ParserFlatbuffersFixture.hpp b/src/armnnTfLiteParser/test/ParserFlatbuffersFixture.hpp index fc1d94e21f..f333ac0d40 100644 --- a/src/armnnTfLiteParser/test/ParserFlatbuffersFixture.hpp +++ b/src/armnnTfLiteParser/test/ParserFlatbuffersFixture.hpp @@ -321,7 +321,8 @@ void ParserFlatbuffersFixture::RunTest(size_t subgraphId, { armnn::BindingPointInfo bindingInfo = m_Parser->GetNetworkOutputBindingInfo(subgraphId, it.first); auto outputExpected = MakeTensor<DataType2, NumOutputDimensions>(bindingInfo.second, it.second, isDynamic); - BOOST_TEST(CompareTensors(outputExpected, outputStorage[it.first], false, isDynamic)); + auto result = CompareTensors(outputExpected, outputStorage[it.first], false, isDynamic); + BOOST_TEST(result.m_Result, result.m_Message.str()); } } @@ -420,6 +421,7 @@ void ParserFlatbuffersFixture::RunTest(size_t subgraphId, { armnn::BindingPointInfo bindingInfo = m_Parser->GetNetworkOutputBindingInfo(subgraphId, it.first); auto outputExpected = MakeTensor<DataType2, NumOutputDimensions>(bindingInfo.second, it.second); - BOOST_TEST(CompareTensors(outputExpected, outputStorage[it.first], false)); + auto result = CompareTensors(outputExpected, outputStorage[it.first], false); + BOOST_TEST(result.m_Result, result.m_Message.str()); } }
\ No newline at end of file |