From a9ac6ba643e8dc4fee88bd0e7e186f0918080c4b Mon Sep 17 00:00:00 2001
From: Mike Kelly <mike.kelly@arm.com>
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 <mike.kelly@arm.com>
Change-Id: I43184cc05e4472011b9347aaa820eb8deb1cd4a0
---
 src/armnn/layers/DetectionPostProcessLayer.cpp | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

(limited to 'src/armnn/layers/DetectionPostProcessLayer.cpp')

diff --git a/src/armnn/layers/DetectionPostProcessLayer.cpp b/src/armnn/layers/DetectionPostProcessLayer.cpp
index 27e459b0ce..6bddf51551 100644
--- a/src/armnn/layers/DetectionPostProcessLayer.cpp
+++ b/src/armnn/layers/DetectionPostProcessLayer.cpp
@@ -50,8 +50,8 @@ void DetectionPostProcessLayer::ValidateTensorShapesFromInputs()
     ARMNN_ASSERT_MSG(GetNumOutputSlots() == 4, "DetectionPostProcessLayer: The layer should return 4 outputs.");
 
     std::vector<TensorShape> inferredShapes = InferOutputShapes(
-            { GetInputSlot(0).GetConnection()->GetTensorInfo().GetShape(),
-              GetInputSlot(1).GetConnection()->GetTensorInfo().GetShape() });
+            { GetInputSlot(0).GetTensorInfo().GetShape(),
+              GetInputSlot(1).GetTensorInfo().GetShape() });
 
     ARMNN_ASSERT(inferredShapes.size() == 4);
     ARMNN_ASSERT(inferredShapes[0].GetDimensionality() == Dimensionality::Specified);
-- 
cgit v1.2.1