diff options
Diffstat (limited to 'runtimes/pure_arm_compute/src/internal/op/ReduceMax.h')
-rw-r--r-- | runtimes/pure_arm_compute/src/internal/op/ReduceMax.h | 45 |
1 files changed, 41 insertions, 4 deletions
diff --git a/runtimes/pure_arm_compute/src/internal/op/ReduceMax.h b/runtimes/pure_arm_compute/src/internal/op/ReduceMax.h index e3278aacf..77d8bd869 100644 --- a/runtimes/pure_arm_compute/src/internal/op/ReduceMax.h +++ b/runtimes/pure_arm_compute/src/internal/op/ReduceMax.h @@ -14,6 +14,12 @@ * limitations under the License. */ +/** + * @file ReduceMax.h + * @ingroup COM_AI_RUNTIME + * @brief This file defines internal::tflite::op::ReduceMax::Param struct + * and internal::tflite::op::ReduceMax::Node class + */ #ifndef __INTERNAL_OP_REDUCEMAX_H__ #define __INTERNAL_OP_REDUCEMAX_H__ @@ -30,32 +36,63 @@ namespace op namespace ReduceMax { +/** + * @brief Struct to have indexes for operation parameter + */ struct Param { - int32_t ofm_index; - - int32_t ifm_index; - int32_t axis_index; + int32_t ofm_index; /**< Index of output feature map */ + int32_t ifm_index; /**< Index of input feature map */ + int32_t axis_index; /**< Index of axis */ + /** + * @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: |