From 526647333571169076f5e72c9fb18c71025bf7c0 Mon Sep 17 00:00:00 2001 From: Teresa Charlin Date: Mon, 29 Jun 2020 16:27:03 +0100 Subject: IVGCVSW-4903 Connect axis parameter in Gather from android to ACL. !android-nn-driver:3302 Signed-off-by: Teresa Charlin Change-Id: Ifbc49acb5272f8a36719bb68676e44817190537d --- src/armnn/layers/GatherLayer.hpp | 11 ++++++----- 1 file changed, 6 insertions(+), 5 deletions(-) (limited to 'src/armnn/layers/GatherLayer.hpp') diff --git a/src/armnn/layers/GatherLayer.hpp b/src/armnn/layers/GatherLayer.hpp index 598ca44dc4..d8737adbee 100644 --- a/src/armnn/layers/GatherLayer.hpp +++ b/src/armnn/layers/GatherLayer.hpp @@ -1,17 +1,17 @@ // -// Copyright © 2017 Arm Ltd. All rights reserved. +// Copyright © 2017 Arm Ltd and Contributors. All rights reserved. // SPDX-License-Identifier: MIT // #pragma once -#include "Layer.hpp" +#include "LayerWithParameters.hpp" namespace armnn { /// This layer represents a Gather operator. -class GatherLayer : public Layer +class GatherLayer : public LayerWithParameters { public: /// Makes a workload for the Gather type. @@ -24,7 +24,7 @@ public: /// @param [in] graph The graph into which this layer is being cloned. GatherLayer* Clone(Graph& graph) const override; - /// Check if the input tensor shape(s) + /// Check if the input tensor shape(s). /// will lead to a valid configuration of @ref GatherLayer. /// @param [in] shapeInferenceMethod Indicates if output shape shall be overwritten or just validate. void ValidateTensorShapesFromInputs( @@ -34,8 +34,9 @@ public: protected: /// Constructor to create a GatherLayer. + /// @param [in] param GatherDescriptor to configure the stack operation. /// @param [in] name Optional name for the layer. - GatherLayer(const char* name); + GatherLayer(const GatherDescriptor& param, const char* name); /// Default destructor ~GatherLayer() = default; -- cgit v1.2.1