1 files changed, 42 insertions, 4 deletions
diff --git a/runtimes/pure_arm_compute/src/internal/op/Split.h b/runtimes/pure_arm_compute/src/internal/op/Split.h
index 8bea1000d..b2c6c2fd1 100644
--- a/runtimes/pure_arm_compute/src/internal/op/Split.h
+++ b/runtimes/pure_arm_compute/src/internal/op/Split.h
@@ -14,6 +14,12 @@
  * limitations under the License.
  */
 
+/**
+ * @file Split.h
+ * @ingroup COM_AI_RUNTIME
+ * @brief This file defines Split node
+ */
+
 #ifndef __INTERNAL_OP_SPLIT_H__
 #define __INTERNAL_OP_SPLIT_H__
 
@@ -31,36 +37,68 @@ namespace op
 namespace Split
 {
 
+/**
+ * @brief Struct to manipulate parameter for Split operation
+ */
 struct Param
 {
-  int32_t axis_index;
-  int32_t ifm_index;
+  int32_t axis_index; //!< index for axis
+  int32_t ifm_index;  //!< index for input feature map
 
-  std::vector<int32_t> ofm_indexes;
+  std::vector<int32_t> ofm_indexes; //!< index for output feature map
 
+  /**
+   * @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 Split Operation
+ */
 class Node final : public op::Node
 {
 public:
+  /**
+   * @brief Construct a new Split Node object
+   * @param param Parameter for Split Node
+   */
   Node(const Param &param) : _param(param)
   {
     // DO NOTHING
   }
 
 public:
+  /**
+   * @brief Default Destructor
+   */
   virtual ~Node() = default;
 
 public:
+  /**
+   * @brief Get parameter
+   * @return Param reference
+   */
   const Param &param(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 Split node
 };
 
 } // namespace Split