From bf7edb619cd5cf0ae84342299abe7c27f3ba6e7d Mon Sep 17 00:00:00 2001 From: mathad01 Date: Tue, 20 Apr 2021 16:12:45 +0100 Subject: IVGCVSW-5418 ExecuteNetwork test for MobileBERT * Removed check in TfLiteParser and Delegate that requires both weights and biases to be constant or non-constant simultaneously * Updated TfLiteParser FullyConnected layer test to properly use non-constant weights * MobileBERT Float32 model now runs on TfLiteParser Signed-off-by: mathad01 Change-Id: I1d75eea466caa90cd695ad353160362df2f69483 --- src/armnnTfLiteParser/test/FullyConnected.cpp | 7 ++----- 1 file changed, 2 insertions(+), 5 deletions(-) (limited to 'src/armnnTfLiteParser/test') diff --git a/src/armnnTfLiteParser/test/FullyConnected.cpp b/src/armnnTfLiteParser/test/FullyConnected.cpp index 333e17fafd..1ce1b2f74f 100644 --- a/src/armnnTfLiteParser/test/FullyConnected.cpp +++ b/src/armnnTfLiteParser/test/FullyConnected.cpp @@ -224,7 +224,7 @@ struct FullyConnectedNonConstWeightsFixture : public ParserFlatbuffersFixture "is_variable": true }, )"; - biasBuffer = R"(,{ "data": [ 10, 0, 0, 0 ] } )"; + biasBuffer = R"(,{ "data": [] } )"; outputs = "3"; } m_JsonString = R"( @@ -250,7 +250,6 @@ struct FullyConnectedNonConstWeightsFixture : public ParserFlatbuffersFixture "details_type": 0, "quantized_dimension": 0 }, - "is_variable": false }, { "shape": )" + filterShape + R"(, @@ -263,7 +262,6 @@ struct FullyConnectedNonConstWeightsFixture : public ParserFlatbuffersFixture "details_type": 0, "quantized_dimension": 0 }, - "is_variable": true }, )" + biasTensor + R"( { @@ -281,7 +279,6 @@ struct FullyConnectedNonConstWeightsFixture : public ParserFlatbuffersFixture "details_type": 0, "quantized_dimension": 0 }, - "is_variable": false } ], "inputs": )" + inputTensors + R"(, @@ -309,7 +306,7 @@ struct FullyConnectedNonConstWeightsFixture : public ParserFlatbuffersFixture "data": [] }, { - "data": [ 2, 3, 4, 5 ] + "data": [] } )" + biasBuffer + R"( ] -- cgit v1.2.1