diff options
Diffstat (limited to 'runtimes/pure_arm_compute/src/internal/op/Logistic.h')
-rw-r--r-- | runtimes/pure_arm_compute/src/internal/op/Logistic.h | 42 |
1 files changed, 39 insertions, 3 deletions
diff --git a/runtimes/pure_arm_compute/src/internal/op/Logistic.h b/runtimes/pure_arm_compute/src/internal/op/Logistic.h index db8935846..a42fdc0d4 100644 --- a/runtimes/pure_arm_compute/src/internal/op/Logistic.h +++ b/runtimes/pure_arm_compute/src/internal/op/Logistic.h @@ -14,6 +14,12 @@ * limitations under the License. */ +/** + * @file Logistic.h + * @ingroup COM_AI_RUNTIME + * @brief This file defines internal::tflite::op::Logistic::Param struct + * and internal::tflite::op::Logistic::Node class + */ #ifndef __INTERNAL_OP_LOGISTIC_H__ #define __INTERNAL_OP_LOGISTIC_H__ @@ -30,31 +36,61 @@ namespace op namespace Logistic { +/** + * @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 ifm_index; /**< Index of input feature map */ + /** + * @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 + * @return N/A + */ void accept(NodeVisitor &&) const override; private: |