diff options
Diffstat (limited to 'src/armnn/layers/ReverseV2Layer.hpp')
-rw-r--r-- | src/armnn/layers/ReverseV2Layer.hpp | 11 |
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; |