diff options
Diffstat (limited to 'src')
-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(); |