From 073d7a3e829dd52f47bcd4e2d0ab4b04cc811b32 Mon Sep 17 00:00:00 2001 From: Jim Flynn Date: Mon, 13 May 2019 13:52:56 +0100 Subject: IVGCVSW-3076 Add ConcatLayer methods to public API !armnn:1119 Change-Id: Ic187f49427e38345f279537ed10a9275980c1360 Signed-off-by: Jim Flynn --- 1.0/HalPolicy.cpp | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/1.0/HalPolicy.cpp b/1.0/HalPolicy.cpp index 158f0e36..fa29f860 100644 --- a/1.0/HalPolicy.cpp +++ b/1.0/HalPolicy.cpp @@ -292,7 +292,7 @@ bool HalPolicy::ConvertConcatenation(const Operation& operation, const Model& mo std::transform(inputHandles.begin(), inputHandles.end(), std::back_inserter(inputTensorInfos), [](const LayerInputHandle& h) -> const armnn::TensorInfo*{ return &h.GetTensorInfo(); }); if (!IsLayerSupportedForAnyBackend(__func__, - armnn::IsMergerSupported, + armnn::IsConcatSupported, data.m_Backends, inputTensorInfos, outputInfo, @@ -301,7 +301,7 @@ bool HalPolicy::ConvertConcatenation(const Operation& operation, const Model& mo return false; } - armnn::IConnectableLayer* layer = data.m_Network->AddMergerLayer(mergerDescriptor); + armnn::IConnectableLayer* layer = data.m_Network->AddConcatLayer(mergerDescriptor); assert(layer != nullptr); layer->GetOutputSlot(0).SetTensorInfo(outputInfo); -- cgit v1.2.1