diff options
author | Jim Flynn <jim.flynn@arm.com> | 2019-05-22 18:00:04 +0100 |
---|---|---|
committer | Matteo Martincigh <matteo.martincigh@arm.com> | 2019-05-29 09:27:40 +0100 |
commit | 7b1e41f454609e6066d0e23f5f74de2f93fe87b4 (patch) | |
tree | 3626ac87f9bd8bdd90d43f7835850ffc9ec51a58 /1.0/HalPolicy.cpp | |
parent | 9972b4012765ba659a4a693d3e918c89f71eda7c (diff) | |
download | android-nn-driver-7b1e41f454609e6066d0e23f5f74de2f93fe87b4.tar.gz |
IVGCVSW-3119 Rename Merger to Concat
!armnn:1209
Change-Id: Ic493e5cdfe479e459342d7c7c9d77c63f859fa30
Signed-off-by: Jim Flynn <jim.flynn@arm.com>
Diffstat (limited to '1.0/HalPolicy.cpp')
-rw-r--r-- | 1.0/HalPolicy.cpp | 14 |
1 files changed, 7 insertions, 7 deletions
diff --git a/1.0/HalPolicy.cpp b/1.0/HalPolicy.cpp index 17d3d3e1..deb31d34 100644 --- a/1.0/HalPolicy.cpp +++ b/1.0/HalPolicy.cpp @@ -265,19 +265,19 @@ bool HalPolicy::ConvertConcatenation(const Operation& operation, const Model& mo // the handles and shapes with the swizzled layer output handles and shapes SwizzleInputs(*data.m_Network, inputHandles, inputShapes, permutationPair.first); - // Create an armnn merger layer descriptor - this will also perform validation on the input shapes - armnn::OriginsDescriptor mergerDescriptor; + // Create an armnn concat layer descriptor - this will also perform validation on the input shapes + armnn::OriginsDescriptor concatDescriptor; try { - // The merger descriptor is always created across the only supported concat dimension + // The concat descriptor is always created across the only supported concat dimension // which is 0, 1 or 3 for a 4-D tensor, or 0 or 2 for a 3-D tensor. - mergerDescriptor = + concatDescriptor = armnn::CreateDescriptorForConcatenation(inputShapes.begin(), inputShapes.end(), concatDim); } catch (const armnn::Exception& error) { - return Fail("%s: Error preparing merger descriptor. %s", __func__, error.what()); + return Fail("%s: Error preparing concat descriptor. %s", __func__, error.what()); } // Validate the output shape is correct given the input shapes based on the @@ -295,12 +295,12 @@ bool HalPolicy::ConvertConcatenation(const Operation& operation, const Model& mo data.m_Backends, inputTensorInfos, outputInfo, - mergerDescriptor)) + concatDescriptor)) { return false; } - armnn::IConnectableLayer* layer = data.m_Network->AddConcatLayer(mergerDescriptor); + armnn::IConnectableLayer* layer = data.m_Network->AddConcatLayer(concatDescriptor); assert(layer != nullptr); layer->GetOutputSlot(0).SetTensorInfo(outputInfo); |