1 files changed, 41 insertions, 2 deletions
diff --git a/runtimes/pure_arm_compute/src/internal/op/ReLU.h b/runtimes/pure_arm_compute/src/internal/op/ReLU.h
index 64dcf2e14..aaa39b523 100644
--- a/runtimes/pure_arm_compute/src/internal/op/ReLU.h
+++ b/runtimes/pure_arm_compute/src/internal/op/ReLU.h
@@ -14,6 +14,12 @@
  * limitations under the License.
  */
 
+/**
+ * @file ReLU.h
+ * @brief This file contains accept function and params for ReLU operation
+ * @ingroup COM_AI_RUNTIME
+ */
+
 #ifndef __INTERNAL_OP_RELU_H__
 #define __INTERNAL_OP_RELU_H__
 
@@ -30,31 +36,64 @@ namespace op
 namespace ReLU
 {
 
+/**
+ * @brief Struct of ReLU operation's param
+ */
 struct Param
 {
-  int32_t ofm_index;
+  int32_t ofm_index; /**< Output format index */
 
-  int32_t ifm_index;
+  int32_t ifm_index; /**< Input format index */
 
+  /**
+   * @brief Construct a new Param object for ReLU as default
+   */
   Param() = default;
+
+  /**
+   * @brief Construct a new Param object for ReLU with params
+   * @param [in] inputCount The number of input
+   * @param [in] inputs Array containing inputs
+   * @param [in] outputCount The number of output
+   * @param [in] outputs Array containing outputs
+   */
   Param(uint32_t inputCount, const uint32_t *inputs, uint32_t outputCount, const uint32_t *outputs);
 };
 
+/**
+ * @brief Class to define operation node for ReLU
+ */
 class Node final : public op::Node
 {
 public:
+  /**
+   * @brief Construct a new Node object for ReLU with param
+   * @param [in] param Parameters for Node
+   */
   Node(const Param &param) : _param(param)
   {
     // DO NOTHING
   }
 
 public:
+  /**
+   * @brief Destroy the Node object for ReLU
+   */
   virtual ~Node() = default;
 
 public:
+  /**
+   * @brief Get parameters for ReLU
+   * @return Parameters of ReLU
+   */
   const Param &param(void) const { return _param; }
 
 public:
+  /**
+   * @brief Function for accepting node for ReLU
+   * @param [in] v Node visitor for invoking visit function of ReLU
+   * @return N/A
+   */
   void accept(NodeVisitor &&) const override;
 
 private: