diff options
author | narpra01 <narumol.prangnawarat@arm.com> | 2018-09-26 16:57:28 +0100 |
---|---|---|
committer | Matthew Bentham <matthew.bentham@arm.com> | 2018-10-12 11:59:53 +0100 |
commit | 96bedf0a0c6b48a884d4ff90192244259e9efb6d (patch) | |
tree | a51ea3ffe3950eaffcf96c2ec5196dd3dc692e6b /1.1/HalPolicy.cpp | |
parent | 7a1518e3a7cd85c0b17eeb58f51ee50a1e85ef9c (diff) | |
download | android-nn-driver-96bedf0a0c6b48a884d4ff90192244259e9efb6d.tar.gz |
IVGCVSW-1201 - Connect input and output tensor to Mean layer
Change-Id: I848db966d249a086cd139c407cefced111aa707a
Diffstat (limited to '1.1/HalPolicy.cpp')
-rw-r--r-- | 1.1/HalPolicy.cpp | 3 |
1 files changed, 3 insertions, 0 deletions
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); } |