diff options
author | Chunseok Lee <chunseok.lee@samsung.com> | 2019-01-08 17:36:34 +0900 |
---|---|---|
committer | Chunseok Lee <chunseok.lee@samsung.com> | 2019-01-08 17:36:34 +0900 |
commit | bd11b24234d7d43dfe05a81c520aa01ffad06e42 (patch) | |
tree | 57d0d4044977e4fa0e50cd9ba40b32006dff19eb /runtimes/pure_arm_compute/src/internal/op/Gather.h | |
parent | 91f4ba45449f700a047a4aeea00b1a7c84e94c75 (diff) | |
download | nnfw-bd11b24234d7d43dfe05a81c520aa01ffad06e42.tar.gz nnfw-bd11b24234d7d43dfe05a81c520aa01ffad06e42.tar.bz2 nnfw-bd11b24234d7d43dfe05a81c520aa01ffad06e42.zip |
Imported Upstream version 0.3upstream/0.3
Diffstat (limited to 'runtimes/pure_arm_compute/src/internal/op/Gather.h')
-rw-r--r-- | runtimes/pure_arm_compute/src/internal/op/Gather.h | 48 |
1 files changed, 43 insertions, 5 deletions
diff --git a/runtimes/pure_arm_compute/src/internal/op/Gather.h b/runtimes/pure_arm_compute/src/internal/op/Gather.h index 5f7fe956f..4470236eb 100644 --- a/runtimes/pure_arm_compute/src/internal/op/Gather.h +++ b/runtimes/pure_arm_compute/src/internal/op/Gather.h @@ -14,6 +14,12 @@ * limitations under the License. */ +/** + * @file Gather.h + * @ingroup COM_AI_RUNTIME + * @brief This file defines Gather operation + */ + #ifndef __INTERNAL_OP_GATHER_H__ #define __INTERNAL_OP_GATHER_H__ @@ -30,37 +36,69 @@ namespace op namespace Gather { +/** + * @brief Struct to manipulate parameter for Gather operation + */ struct Param { - int32_t ofm_index; // output + int32_t ofm_index; //!< index for output feature map - int32_t lhs_index; // input - int32_t rhs_index; // indexes - int32_t axis_index; // axis + int32_t lhs_index; //!< index for lhs tensor + int32_t rhs_index; //!< index for rhs tensor + int32_t axis_index; //!< index for axis + /** + * @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 Gather Operation + */ class Node final : public op::Node { public: + /** + * @brief Construct a new Tanh Node object + * @param param Parameter for Tanh 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 Gather node }; } // namespace Gather |