From 019840d1161738aefd6ebd32ccf4e72e618cae15 Mon Sep 17 00:00:00 2001 From: Finn Williams Date: Mon, 30 Nov 2020 17:43:28 +0000 Subject: IVGCVSW-5374 Provide an Android build for the delegate Signed-off-by: Finn Williams Change-Id: I33eb8c650be654ad891afd2295f2057f13a9d084 --- delegate/src/Split.hpp | 6 ++---- 1 file changed, 2 insertions(+), 4 deletions(-) (limited to 'delegate/src/Split.hpp') diff --git a/delegate/src/Split.hpp b/delegate/src/Split.hpp index c76ad60ace..8248be9413 100644 --- a/delegate/src/Split.hpp +++ b/delegate/src/Split.hpp @@ -42,10 +42,9 @@ TfLiteStatus VisitSplitOperator(DelegateData& delegateData, return kTfLiteError; } - const armnn::TensorInfo& axisTensorInfo = GetTensorInfoForTfLiteTensor(tfLiteAxisTensor); const armnn::TensorInfo& inputTensorInfo = GetTensorInfoForTfLiteTensor(tfLiteInputTensor); - ARMNN_ASSERT(axisTensorInfo.GetNumElements() == 1); + ARMNN_ASSERT(GetTensorInfoForTfLiteTensor(tfLiteAxisTensor).GetNumElements() == 1); auto* axisTensorDataPtr = tflite::GetTensorData(&tfLiteAxisTensor); std::vector axisTensorData(axisTensorDataPtr, axisTensorDataPtr + 1); const unsigned int splitDim = axisTensorData[0]; @@ -168,9 +167,8 @@ TfLiteStatus VisitSplitVOperator(DelegateData& delegateData, const armnn::TensorInfo& inputTensorInfo = GetTensorInfoForTfLiteTensor(tfLiteInputTensor); const armnn::TensorInfo& splitsTensorInfo = GetTensorInfoForTfLiteTensor(tfLiteSplitsTensor); ARMNN_ASSERT(splitsTensorInfo.GetNumDimensions() == 1); + ARMNN_ASSERT(GetTensorInfoForTfLiteTensor(tfLiteAxisTensor).GetNumElements() == 1); - const armnn::TensorInfo& axisTensorInfo = GetTensorInfoForTfLiteTensor(tfLiteAxisTensor); - ARMNN_ASSERT(axisTensorInfo.GetNumElements() == 1); auto* axisTensorDataPtr = tflite::GetTensorData(&tfLiteAxisTensor); std::vector axisTensorData(axisTensorDataPtr, axisTensorDataPtr + 1); -- cgit v1.2.1