From a9ac6ba643e8dc4fee88bd0e7e186f0918080c4b Mon Sep 17 00:00:00 2001 From: Mike Kelly Date: Fri, 30 Jun 2023 15:18:26 +0100 Subject: IVGCVSW-7828 Add an Optional TensorInfo to InputSlot * Updated calls to use the new function From: GetInputSlot(n).GetConnection()->GetTensorInfo(); To: GetInputSlot(n).GetTensorInfo(); * Added UnitTests Signed-off-by: Mike Kelly Change-Id: I43184cc05e4472011b9347aaa820eb8deb1cd4a0 --- src/armnn/layers/StackLayer.cpp | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) (limited to 'src/armnn/layers/StackLayer.cpp') diff --git a/src/armnn/layers/StackLayer.cpp b/src/armnn/layers/StackLayer.cpp index 3ebacaf3b5..3c5a216eb3 100644 --- a/src/armnn/layers/StackLayer.cpp +++ b/src/armnn/layers/StackLayer.cpp @@ -1,5 +1,5 @@ // -// Copyright © 2017 Arm Ltd and Contributors. All rights reserved. +// Copyright © 2019-2023 Arm Ltd and Contributors. All rights reserved. // SPDX-License-Identifier: MIT // #include "StackLayer.hpp" @@ -78,7 +78,7 @@ void StackLayer::ValidateTensorShapesFromInputs() std::vector inputShapes; for (unsigned int i = 0; i < GetNumInputSlots(); ++i) { - TensorShape inputShape = GetInputSlot(i).GetConnection()->GetTensorInfo().GetShape(); + TensorShape inputShape = GetInputSlot(i).GetTensorInfo().GetShape(); if (inputShape != m_Param.m_InputShape) { throw LayerValidationException("StackLayer: TensorShape set on InputSlot[" + -- cgit v1.2.1