1 files changed, 21 insertions, 2 deletions
diff --git a/runtimes/pure_arm_compute/src/internal/Tensor3DSink.h b/runtimes/pure_arm_compute/src/internal/Tensor3DSink.h
index 20de3b9e8..1e14e2d6c 100644
--- a/runtimes/pure_arm_compute/src/internal/Tensor3DSink.h
+++ b/runtimes/pure_arm_compute/src/internal/Tensor3DSink.h
@@ -14,6 +14,11 @@
  * limitations under the License.
  */
 
+/**
+ * @file    Tensor3DSink.h
+ * @ingroup COM_AI_RUNTIME
+ * @brief   This file defines Tensor3DSink class
+ */
 #ifndef __TENSOR3D_SINK_H__
 #define __TENSOR3D_SINK_H__
 
@@ -26,16 +31,30 @@
 #include <arm_compute/core/Window.h>
 #include <arm_compute/core/Helpers.h>
 
+/**
+ * @brief Class to get tensor data from arm compute tensor
+ */
 template <typename T> class Tensor3DSink final : public Sink
 {
 public:
-  Tensor3DSink(const nnfw::util::tensor::Shape &shape, T *base, const size_t size)
+  /**
+   * @brief     Construct a new Tensor3DSink object
+   * @param[in] shape Shape of tensor
+   * @param[in] base  Pointer to get data
+   * @param[in] size  Size of tensor
+   */
+  Tensor3DSink(const nnfw::misc::tensor::Shape &shape, T *base, const size_t size)
       : _shape{shape}, _base{base}, _size{size}
   {
     // DO NOTHING
   }
 
 public:
+  /**
+   * @brief     Get tensor data from arm compute tensor to base
+   * @param[in] tensor  Tensor object of arm compute to get data
+   * @return    N/A
+   */
   void pull(::arm_compute::ITensor &tensor) const override
   {
     using ::arm_compute::Window;
@@ -60,7 +79,7 @@ public:
   }
 
 private:
-  const nnfw::util::tensor::Shape _shape;
+  const nnfw::misc::tensor::Shape _shape;
 
 private:
   T *const _base;