From 7b89992e740cf9b34f45a679e5e293a85ff75bcf Mon Sep 17 00:00:00 2001 From: Mike Kelly Date: Mon, 17 Jul 2023 14:17:52 +0100 Subject: IVGCVSW-7890 SplitterLayer does not use overridden TensorInfos correctly * The SplitterLayer did not use the Overridden TensorInfos when calculating whether or not to use SubTensors. Signed-off-by: Mike Kelly Change-Id: I675b27c546c6ed720c76f4b9d868ebe32f914c70 --- src/armnn/layers/SplitterLayer.cpp | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/src/armnn/layers/SplitterLayer.cpp b/src/armnn/layers/SplitterLayer.cpp index dc8864a736..2e75139fb4 100644 --- a/src/armnn/layers/SplitterLayer.cpp +++ b/src/armnn/layers/SplitterLayer.cpp @@ -48,8 +48,7 @@ void SplitterLayer::CreateTensors(const TensorHandleFactoryRegistry& registry, // Get outputHandler of previous layer const OutputHandler& outputHandler = GetInputSlots()[0].GetConnectedOutputSlot()->GetOutputHandler(); const OutputSlot* slot = GetInputSlots()[0].GetConnectedOutputSlot(); - - const TensorInfo& parentInfo = outputHandler.GetTensorInfo(); + const TensorInfo& parentInfo = GetInputSlot(0).GetTensorInfo(); ITensorHandle* inputData = outputHandler.GetData(); -- cgit v1.2.1