aboutsummaryrefslogtreecommitdiff
path: root/src/armnn/layers/ReverseV2Layer.hpp
diff options
context:
space:
mode:
Diffstat (limited to 'src/armnn/layers/ReverseV2Layer.hpp')
-rw-r--r--src/armnn/layers/ReverseV2Layer.hpp11
1 files changed, 5 insertions, 6 deletions
diff --git a/src/armnn/layers/ReverseV2Layer.hpp b/src/armnn/layers/ReverseV2Layer.hpp
index 046670e9de..7dc0656aca 100644
--- a/src/armnn/layers/ReverseV2Layer.hpp
+++ b/src/armnn/layers/ReverseV2Layer.hpp
@@ -11,7 +11,7 @@ namespace armnn
{
/// This layer represents a ReverseV2 operation.
- class ReverseV2Layer : public LayerWithParameters<ReverseV2Descriptor>
+ class ReverseV2Layer : public Layer
{
public:
/// Makes a workload for the ReverseV2 type.
@@ -28,19 +28,18 @@ namespace armnn
/// otherwise infers the output shapes from given input shapes and layer properties.
/// @param [in] inputShapes The vector of input shapes for ReverseV2.
/// @return A vector to the inferred output shape.
-
- /// Use the default Layer::InferOutputShape method
- // std::vector<TensorShape> InferOutputShapes(const std::vector<TensorShape>& inputShapes) const override;
+ std::vector<TensorShape> InferOutputShapes(const std::vector<TensorShape>& inputShapes) const override;
/// Check if the input tensor shape(s)
/// will lead to a valid configuration of @ref ReverseV2Layer.
void ValidateTensorShapesFromInputs() override;
+ void ExecuteStrategy(IStrategy& strategy) const override;
+
protected:
/// Constructor to create a ReverseV2Layer.
- /// @param [in] param ReverseV2Descriptor to configure the ReverseV2 operation.
/// @param [in] name Optional name for the layer.
- ReverseV2Layer(const ReverseV2Descriptor& param, const char* name);
+ ReverseV2Layer(const char* name);
/// Default destructor
~ReverseV2Layer() = default;