diff options
Diffstat (limited to 'runtimes/pure_arm_compute/src/internal/op/ResizeBilinear.h')
-rw-r--r-- | runtimes/pure_arm_compute/src/internal/op/ResizeBilinear.h | 47 |
1 files changed, 42 insertions, 5 deletions
diff --git a/runtimes/pure_arm_compute/src/internal/op/ResizeBilinear.h b/runtimes/pure_arm_compute/src/internal/op/ResizeBilinear.h index bf216b75f..f2eab4aaf 100644 --- a/runtimes/pure_arm_compute/src/internal/op/ResizeBilinear.h +++ b/runtimes/pure_arm_compute/src/internal/op/ResizeBilinear.h @@ -14,6 +14,12 @@ * limitations under the License. */ +/** + * @file ResizeBilinear.h + * @ingroup COM_AI_RUNTIME + * @brief This file defines internal::tflite::op::ResizeBilinear::Param struct + * and internal::tflite::op::ResizeBilinear::Node class + */ #ifndef __INTERNAL_OP_RESIZE_BILINEAR_H__ #define __INTERNAL_OP_RESIZE_BILINEAR_H__ @@ -30,33 +36,64 @@ namespace op namespace ResizeBilinear { +/** + * @brief Struct to have indexes for ResizeBilinear operation parameter + */ struct Param { - int32_t ofm_index; - - int32_t ifm_index; - int32_t height_index; - int32_t width_index; + int32_t ofm_index; /**< Index of output feature map */ + int32_t ifm_index; /**< Index of input feature map */ + int32_t height_index; /**< Index of height */ + int32_t width_index; /**< Index of width */ + /** + * @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 ResizeBilinear 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: |