From 96bedf0a0c6b48a884d4ff90192244259e9efb6d Mon Sep 17 00:00:00 2001 From: narpra01 Date: Wed, 26 Sep 2018 16:57:28 +0100 Subject: IVGCVSW-1201 - Connect input and output tensor to Mean layer Change-Id: I848db966d249a086cd139c407cefced111aa707a --- 1.1/HalPolicy.cpp | 3 +++ 1 file changed, 3 insertions(+) (limited to '1.1') diff --git a/1.1/HalPolicy.cpp b/1.1/HalPolicy.cpp index e3ccf73c..de743088 100644 --- a/1.1/HalPolicy.cpp +++ b/1.1/HalPolicy.cpp @@ -196,6 +196,9 @@ bool HalPolicy::ConvertMean(const Operation& operation, const Model& model, Conv } 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); } -- cgit v1.2.1