GitHub #723 Fix Maybe uninitialized variable
Signed-off-by: Teresa Charlin <teresa.charlinreyes@arm.com>
Change-Id: I70d3673bbb2dc113ac5eb6d63029a4a4a684e831
diff --git a/include/armnn/backends/Workload.hpp b/include/armnn/backends/Workload.hpp
index be6fbd9..9f7aad0 100644
--- a/include/armnn/backends/Workload.hpp
+++ b/include/armnn/backends/Workload.hpp
@@ -1,5 +1,5 @@
//
-// Copyright © 2022 Arm Ltd and Contributors. All rights reserved.
+// Copyright © 2022-2023 Arm Ltd and Contributors. All rights reserved.
// SPDX-License-Identifier: MIT
//
#pragma once
@@ -124,6 +124,8 @@
if (!info.m_InputTensorInfos.empty())
{
+ expectedInputType = info.m_InputTensorInfos.front().GetDataType();
+
if (expectedOutputType != expectedInputType)
{
ARMNN_ASSERT_MSG(false, "Trying to create workload with incorrect type");