diff options
Diffstat (limited to 'runtimes/pure_arm_compute/src/internal/op/AvgPool2D.h')
| -rw-r--r-- | runtimes/pure_arm_compute/src/internal/op/AvgPool2D.h | 110 |
1 files changed, 89 insertions, 21 deletions
diff --git a/runtimes/pure_arm_compute/src/internal/op/AvgPool2D.h b/runtimes/pure_arm_compute/src/internal/op/AvgPool2D.h index 729f6043c..cf9061ca9 100644 --- a/runtimes/pure_arm_compute/src/internal/op/AvgPool2D.h +++ b/runtimes/pure_arm_compute/src/internal/op/AvgPool2D.h @@ -14,6 +14,12 @@ * limitations under the License. */ +/** + * @file AvgPool2D.h + * @ingroup COM_AI_RUNTIME + * @brief This file defines internal::tflite::op::AvgPool2D Param structs + * and internal::tflite::op::AvgPool2D Node classes + */ #ifndef __INTERNAL_OP_AVG_POOL_2D_H__ #define __INTERNAL_OP_AVG_POOL_2D_H__ @@ -32,44 +38,75 @@ namespace AvgPool2D namespace Explicit { +/** + * @brief Struct to have indexes for operation parameter + */ struct Param { - int32_t ofm_index; + int32_t ofm_index; /**< Index of output feature map */ - int32_t ifm_index; + int32_t ifm_index; /**< Index of input feature map */ - int32_t kw_index; - int32_t kh_index; + int32_t kw_index; /**< Index of kernel width */ + int32_t kh_index; /**< Index of kernel height */ - int32_t hstride_index; - int32_t vstride_index; + int32_t hstride_index; /**< Index of horizontal stride */ + int32_t vstride_index; /**< Index of vertical stride */ - int32_t padding_left_index; - int32_t padding_right_index; - int32_t padding_top_index; - int32_t padding_bottom_index; - - int32_t activation_index; + int32_t padding_left_index; /**< Index of padding left */ + int32_t padding_right_index; /**< Index of padding right */ + int32_t padding_top_index; /**< Index of padding top */ + int32_t padding_bottom_index; /**< Index of padding bottom */ + int32_t activation_index; /**< Index of activation */ + /** + * @brief Construct as default + */ Param() = default; + /** + * @brief Construct a new Param object with params + * @param[in] inputCount Count of inputs + * @param[in] inputs Pointer of inputs + * @param[in] outputCount Count of outputs + * @param[in] outputs Pointer of outputs + */ Param(uint32_t inputCount, const uint32_t *inputs, uint32_t outputCount, const uint32_t *outputs); }; +/** + * @brief Class to represent an operation of data structure + */ class Node final : public op::Node { public: + /** + * @brief Construct a new Node object with param + * @param[in] param Param object that makes up a Node + */ Node(const Param ¶m) : _param(param) { // DO NOTHING } public: + /** + * @brief Destruct as default + */ virtual ~Node() = default; public: + /** + * @brief Get a reference of Param object + * @return Reference of Param object + */ const Param ¶m(void) const { return _param; } public: + /** + * @brief Visit this Node by NodeVisitor + * @param[in] v Visitor + * @return N/A + */ void accept(NodeVisitor &&) const override; private: @@ -81,40 +118,71 @@ private: namespace Implicit { +/** + * @brief Struct to have indexes for operation parameter + */ struct Param { - int32_t ofm_index; - - int32_t ifm_index; + int32_t ofm_index; /**< Index of output feature map */ - int32_t kw_index; - int32_t kh_index; + int32_t ifm_index; /**< Index of input feature map */ - int32_t hstride_index; - int32_t vstride_index; + int32_t kw_index; /**< Index of kernel width */ + int32_t kh_index; /**< Index of kernel height */ - int32_t padding_index; - int32_t activation_index; + int32_t hstride_index; /**< Index of horizontal stride */ + int32_t vstride_index; /**< Index of vertical stride */ + int32_t padding_index; /**< Index of padding */ + int32_t activation_index; /**< Index of activation */ + /** + * @brief Construct as default + */ Param() = default; + /** + * @brief Construct a new Param object with params + * @param[in] inputCount Count of inputs + * @param[in] inputs Pointer of inputs + * @param[in] outputCount Count of outputs + * @param[in] outputs Pointer of outputs + */ Param(uint32_t inputCount, const uint32_t *inputs, uint32_t outputCount, const uint32_t *outputs); }; +/** + * @brief Class to represent an operation of data structure + */ class Node final : public op::Node { public: + /** + * @brief Construct a new Node object with param + * @param[in] param Param object that makes up a Node + */ Node(const Param ¶m) : _param(param) { // DO NOTHING } public: + /** + * @brief Destruct as default + */ virtual ~Node() = default; public: + /** + * @brief Get a reference of Param object + * @return Reference of Param object + */ const Param ¶m(void) const { return _param; } public: + /** + * @brief Visit this Node by NodeVisitor + * @param[in] v Visitor + * @return N/A + */ void accept(NodeVisitor &&) const override; private: |