commit 96bedf0a0c6b48a884d4ff90192244259e9efb6d
author narpra01 <narumol.prangnawarat@arm.com> Wed Sep 26 16:57:28 2018 +0100
committer Matthew Bentham <matthew.bentham@arm.com> Fri Oct 12 11:59:53 2018 +0100
tree a51ea3ffe3950eaffcf96c2ec5196dd3dc692e6b
parent 7a1518e3a7cd85c0b17eeb58f51ee50a1e85ef9c

IVGCVSW-1201 - Connect input and output tensor to Mean layer

Change-Id: I848db966d249a086cd139c407cefced111aa707a

1.1/HalPolicy.cpp

diff --git a/1.1/HalPolicy.cpp b/1.1/HalPolicy.cpp
index e3ccf73..de74308 100644
--- a/1.1/HalPolicy.cpp
+++ b/1.1/HalPolicy.cpp
@@ -196,6 +196,9 @@
     }
 
     armnn::IConnectableLayer* const layer = data.m_Network->AddMeanLayer(descriptor);
+    assert(layer != nullptr);
+    input.Connect(layer->GetInputSlot(0));
+    layer->GetOutputSlot(0).SetTensorInfo(outputInfo);
 
     return SetupAndTrackLayerOutputSlot(operation, 0, *layer, model, data);
 }