diff options
author | Mike Kelly <mike.kelly@arm.com> | 2023-07-17 14:17:52 +0100 |
---|---|---|
committer | TeresaARM <teresa.charlinreyes@arm.com> | 2023-07-17 15:12:40 +0000 |
commit | 7b89992e740cf9b34f45a679e5e293a85ff75bcf (patch) | |
tree | f88548925e7ae1d29d379e232691b179520bfb86 | |
parent | bb8d7591a35bd95480b39001f8b7e41a6671f3a6 (diff) | |
download | armnn-7b89992e740cf9b34f45a679e5e293a85ff75bcf.tar.gz |
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 <mike.kelly@arm.com>
Change-Id: I675b27c546c6ed720c76f4b9d868ebe32f914c70
-rw-r--r-- | src/armnn/layers/SplitterLayer.cpp | 3 |
1 files changed, 1 insertions, 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(); |