diff options
author | mathad01 <matthew.haddon@arm.com> | 2021-04-20 16:12:45 +0100 |
---|---|---|
committer | Jan Eilers <jan.eilers@arm.com> | 2021-04-22 08:21:43 +0000 |
commit | bf7edb619cd5cf0ae84342299abe7c27f3ba6e7d (patch) | |
tree | aace117c1b6de99b257552be74b8d8dc2b39718d /delegate/src | |
parent | e11e63d749b0909f13f9a39c8d34ef5523255170 (diff) | |
download | armnn-bf7edb619cd5cf0ae84342299abe7c27f3ba6e7d.tar.gz |
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 <matthew.haddon@arm.com>
Change-Id: I1d75eea466caa90cd695ad353160362df2f69483
Diffstat (limited to 'delegate/src')
-rw-r--r-- | delegate/src/FullyConnected.hpp | 9 |
1 files changed, 0 insertions, 9 deletions
diff --git a/delegate/src/FullyConnected.hpp b/delegate/src/FullyConnected.hpp index 2b45c48a89..e94304fb21 100644 --- a/delegate/src/FullyConnected.hpp +++ b/delegate/src/FullyConnected.hpp @@ -77,15 +77,6 @@ TfLiteStatus VisitFullyConnectedOperator(DelegateData& delegateData, { return kTfLiteError; } - if ((isConstantWeights && !tflite::IsConstantTensor(&tfLiteBiasTensor)) - || (!isConstantWeights && tflite::IsConstantTensor(&tfLiteBiasTensor))) - { - TF_LITE_MAYBE_KERNEL_LOG( - tfLiteContext, - "TfLiteArmnnDelegate: Weights and bias are not compatible" - " in operator #%d node #%d: ", operatorCode, nodeIndex); - return kTfLiteError; - } biasTensorInfo = GetTensorInfoForTfLiteTensor(tfLiteBiasTensor); } else |