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 --- delegate/src/FullyConnected.hpp | 9 --------- 1 file changed, 9 deletions(-) (limited to 'delegate/src/FullyConnected.hpp') 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 -- cgit v1.2.1