diff options
Diffstat (limited to 'runtimes/pure_arm_compute/src/internal/op/Reshape.h')
-rw-r--r-- | runtimes/pure_arm_compute/src/internal/op/Reshape.h | 46 |
1 files changed, 42 insertions, 4 deletions
diff --git a/runtimes/pure_arm_compute/src/internal/op/Reshape.h b/runtimes/pure_arm_compute/src/internal/op/Reshape.h index ab77ade8c..7152eaece 100644 --- a/runtimes/pure_arm_compute/src/internal/op/Reshape.h +++ b/runtimes/pure_arm_compute/src/internal/op/Reshape.h @@ -14,6 +14,12 @@ * limitations under the License. */ +/** + * @file Reshape.h + * @ingroup COM_AI_RUNTIME + * @brief This file defines Reshape node + */ + #ifndef __INTERNAL_OP_RESHAPE_H__ #define __INTERNAL_OP_RESHAPE_H__ @@ -30,36 +36,68 @@ namespace op namespace Reshape { +/** + * @brief Struct to manipulate parameter for Reshape operation + */ struct Param { - int32_t output_index; + int32_t output_index; //!< index for output feature map - int32_t input_index; - int32_t shape_index; + int32_t input_index; //!< index for input feature map + int32_t shape_index; //!< index for shape + /** + * @brief Default Constructor + */ Param() = default; + /** + * @brief Construct a new Param object + * @param[in] inputCount the number of inputs + * @param[in] inputs pointer for input data + * @param[in] outputCount the number of outputs + * @param[in] outputs pointer for input data + */ Param(uint32_t inputCount, const uint32_t *inputs, uint32_t outputCount, const uint32_t *outputs); }; +/** + * @brief Class to define Reshape Operation + */ class Node final : public op::Node { public: + /** + * @brief Construct a new Reshape Node object + * @param param Parameter for Reshape Node + */ Node(const Param ¶m) : _param(param) { // DO NOTHING } public: + /** + * @brief Default Destructor + */ virtual ~Node() = default; public: + /** + * @brief Get parameter + * @return Param reference + */ const Param ¶m(void) const { return _param; } public: + /** + * @brief Accept a NodeVisitor so that it can visit this node + * @param [in] v Visitor + * @return N/A + */ void accept(NodeVisitor &&) const override; private: - const Param _param; + const Param _param; //!< parameter for Reshape node }; } // namespace Reshape |