patch/23.02/_reduce_layer_8hpp_source.xhtml

 //
 // Copyright © 2020 Samsung Electronics Co Ltd and Contributors. All rights reserved.
 // Copyright © 2023 Arm Ltd and Contributors. All rights reserved.
 // SPDX-License-Identifier: MIT
 //
 #pragma once

 #include "LayerWithParameters.hpp"

 namespace armnn
 {

 /// This layer represents a reduction operation.
 class ReduceLayer : public LayerWithParameters<ReduceDescriptor>
 {
 public:
     /// Makes a workload for the Reduce type.
     /// @param [in] graph The graph where this layer can be found.
     /// @param [in] factory The workload factory which will create the workload.
     /// @return A pointer to the created workload, or nullptr if not created.
     virtual std::unique_ptr<IWorkload>CreateWorkload(const IWorkloadFactory& factory) const override;

     /// Creates a dynamically-allocated copy of this layer.
     /// @param [in] graph The graph into which this layer is being cloned.
     ReduceLayer* Clone(Graph& graph) const override;

     /// By default returns inputShapes if the number of inputs are equal to number of outputs,
     /// otherwise infers the output shapes from given input shapes and layer properties.
     /// @param [in] inputShapes The input shapes layer has.
     /// @return A vector to the inferred output shape.
     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 ReduceLayer.
     void ValidateTensorShapesFromInputs() override;

     void ExecuteStrategy(IStrategy& strategy) const override;

 protected:
     /// Constructor to create a ReduceLayer.
     /// @param [in] param ReduceDescriptor to configure the reduction operation.
     /// @param [in] name Optional name for the layer.
     ReduceLayer(const ReduceDescriptor& param, const char* name);

     /// Default destructor
     ~ReduceLayer() = default;
 };

 } // namespace armnn
armnn::ReduceLayer::~ReduceLayer
~ReduceLayer()=default
Default destructor.

-LayerWithParameters.hpp

-armnn::IWorkloadFactory
Definition: WorkloadFactory.hpp:22

-armnn::ReduceLayer::ValidateTensorShapesFromInputs
void ValidateTensorShapesFromInputs() override
Check if the input tensor shape(s) will lead to a valid configuration of ReduceLayer.
Definition: ReduceLayer.cpp:44

-armnn::ReduceLayer::CreateWorkload
virtual std::unique_ptr< IWorkload > CreateWorkload(const IWorkloadFactory &factory) const override
Makes a workload for the Reduce type.
Definition: ReduceLayer.cpp:23

-armnn
Copyright (c) 2021 ARM Limited and Contributors.
Definition: 01_00_quick_start.dox:6

-armnn::ReduceLayer
This layer represents a reduction operation.
Definition: ReduceLayer.hpp:14

-armnn::ReduceLayer::ReduceLayer
ReduceLayer(const ReduceDescriptor &param, const char *name)
Constructor to create a ReduceLayer.
Definition: ReduceLayer.cpp:18

-armnn::IStrategy
Definition: IStrategy.hpp:16

-armnn::ReduceDescriptor
A ReduceDescriptor for the REDUCE operators.
Definition: Descriptors.hpp:1485

-armnn::Graph
Definition: Graph.hpp:30

-armnn::ReduceLayer::ExecuteStrategy
void ExecuteStrategy(IStrategy &strategy) const override
Apply a visitor to this layer.
Definition: ReduceLayer.cpp:117

-armnn::LayerWithParameters
Definition: LayerWithParameters.hpp:14

-armnn::ReduceLayer::Clone
ReduceLayer * Clone(Graph &graph) const override
Creates a dynamically-allocated copy of this layer.
Definition: ReduceLayer.cpp:34

-armnn::ReduceLayer::InferOutputShapes
std::vector< TensorShape > InferOutputShapes(const std::vector< TensorShape > &inputShapes) const override
By default returns inputShapes if the number of inputs are equal to number of outputs, otherwise infers the output shapes from given input shapes and layer properties.
Definition: ReduceLayer.cpp:62

+Go to the documentation of this file. //

+ // Copyright © 2020 Samsung Electronics Co Ltd and Contributors. All rights reserved.

+ // Copyright © 2023 Arm Ltd and Contributors. All rights reserved.

+ // SPDX-License-Identifier: MIT

+ //

+ #pragma once

+

+ #include "LayerWithParameters.hpp"

+

+ namespace armnn

+ {

+

+ /// This layer represents a reduction operation.

+ class ReduceLayer : public LayerWithParameters<ReduceDescriptor>

+ {

+ public:

+     /// Makes a workload for the Reduce type.

+     /// @param [in] graph The graph where this layer can be found.

+     /// @param [in] factory The workload factory which will create the workload.

+     /// @return A pointer to the created workload, or nullptr if not created.

+     virtual std::unique_ptr<IWorkload>CreateWorkload(const IWorkloadFactory& factory) const override;

+

+     /// Creates a dynamically-allocated copy of this layer.

+     /// @param [in] graph The graph into which this layer is being cloned.

+     ReduceLayer* Clone(Graph& graph) const override;

+

+     /// By default returns inputShapes if the number of inputs are equal to number of outputs,

+     /// otherwise infers the output shapes from given input shapes and layer properties.

+     /// @param [in] inputShapes The input shapes layer has.

+     /// @return A vector to the inferred output shape.

+     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 ReduceLayer.

+     void ValidateTensorShapesFromInputs() override;

+

+     void ExecuteStrategy(IStrategy& strategy) const override;

+

+ protected:

+     /// Constructor to create a ReduceLayer.

+     /// @param [in] param ReduceDescriptor to configure the reduction operation.

+     /// @param [in] name Optional name for the layer.

+     ReduceLayer(const ReduceDescriptor& param, const char* name);

+

+     /// Default destructor

+     ~ReduceLayer() = default;

+ };

+

+ } // namespace armnn