diff --git a/Applications/SimpleFC/jni/main.cpp b/Applications/SimpleFC/jni/main.cpp
index 30626bf2b..cddde9868 100644
--- a/Applications/SimpleFC/jni/main.cpp
+++ b/Applications/SimpleFC/jni/main.cpp
@@ -131,8 +131,8 @@ void createAndRun(unsigned int epochs, unsigned int batch_size,
 
   float input[1 * 1024];
 
-  for (uint j = 0; j < feature_size; ++j)
-    input[j] = (j / feature_size);
+  for (unsigned int j = 0; j < feature_size; ++j)
+    input[j] = (j / (float)feature_size);
 
   std::vector<float *> in;
   std::vector<float *> answer;
@@ -183,8 +183,8 @@ int main(int argc, char *argv[]) {
   std::cout << "swap_on : " << swap_on << std::endl;
   std::cout << "look_ahead : " << look_ahead << std::endl;
 
-  uint batch_size = 1;
-  uint epoch = 1;
+  unsigned int batch_size = 1;
+  unsigned int epoch = 1;
 
   try {
     createAndRun(epoch, batch_size, swap_on, look_ahead);
diff --git a/nntrainer/graph/network_graph.h b/nntrainer/graph/network_graph.h
index 3ba08d05c..b2398f1f4 100644
--- a/nntrainer/graph/network_graph.h
+++ b/nntrainer/graph/network_graph.h
@@ -502,7 +502,18 @@ class NetworkGraph {
    */
   bool isMixedPrecision() { return (!istrequal(tensor_dtype[1], "FP32")); }
 
+  /**
+   * @brief Get Number of Loaded WeightPool Tensor
+   *
+   * @return Number of Loaded WeightPool Tensor
+   */
   unsigned int getNumLoadedWeightPoolTensors();
+
+  /**
+   * @brief Get Number of Loaded TensorPool Tensor
+   *
+   * @return Number of Loaded TensorPool Tensor
+   */
   unsigned int getNumLoadedTensorPoolTensors();
 
 private:
diff --git a/nntrainer/layers/layer_node.h b/nntrainer/layers/layer_node.h
index 0513c485a..9db427b33 100644
--- a/nntrainer/layers/layer_node.h
+++ b/nntrainer/layers/layer_node.h
@@ -750,7 +750,9 @@ class LayerNode final : public ml::train::Layer, public GraphNode {
   /**
    * @brief     read layer Weight & Bias data from file
    * @param file input file stream
-   * @param bool read optimizer variables
+   * @param swap swap type
+   * @param mode Execution mode
+   * @param opt_var read optimizer variables
    */
   void read(std::ifstream &file, bool opt_var = false,
             ml::train::ExecutionMode mode = ml::train::ExecutionMode::TRAIN,
diff --git a/nntrainer/tensor/cache_loader.h b/nntrainer/tensor/cache_loader.h
index ce13c455a..a58b67b61 100644
--- a/nntrainer/tensor/cache_loader.h
+++ b/nntrainer/tensor/cache_loader.h
@@ -115,6 +115,11 @@ class CacheLoader {
    */
   virtual int cancelAsync(int id);
 
+  /**
+   * @brief Get number of loaded tensors
+   *
+   * @return number of loaded tensors
+   */
   virtual unsigned int getNumLoadedTensors();
 
 private:
diff --git a/nntrainer/tensor/cache_pool.h b/nntrainer/tensor/cache_pool.h
index 89aa6715a..4a70db27c 100644
--- a/nntrainer/tensor/cache_pool.h
+++ b/nntrainer/tensor/cache_pool.h
@@ -188,6 +188,11 @@ class CachePool : public MemoryPool {
    */
   virtual std::string getName() { return name; }
 
+  /**
+   * @brief Get number of loaded tensors
+   *
+   * @return number of loaded tensors
+   */
   virtual unsigned int getNumLoadedTensors();
 
 protected:
diff --git a/nntrainer/tensor/manager.h b/nntrainer/tensor/manager.h
index d0f9164e4..df9ae622f 100644
--- a/nntrainer/tensor/manager.h
+++ b/nntrainer/tensor/manager.h
@@ -540,8 +540,18 @@ class Manager {
    */
   bool isMixedPrecision() { return !istrequal(tensor_dtype[0], "FP32"); }
 
+  /**
+   * @brief Get Number of Loaded WeightPool Tensor
+   *
+   * @return Number of Loaded WeightPool Tensor
+   */
   unsigned int getNumLoadedWeightPoolTensors();
 
+  /**
+   * @brief Get Number of Loaded TensorPool Tensor
+   *
+   * @return Number of Loaded TensorPool Tensor
+   */
   unsigned int getNumLoadedTensorPoolTensors();
 
 private:
diff --git a/nntrainer/tensor/swap_device.h b/nntrainer/tensor/swap_device.h
index 9a64b58eb..89020d67a 100644
--- a/nntrainer/tensor/swap_device.h
+++ b/nntrainer/tensor/swap_device.h
@@ -119,6 +119,11 @@ class SwapDevice {
    */
   const std::string getDevicePath() const { return dev_path; }
 
+  /**
+   * @brief Get number of loaded tensors
+   *
+   * @return number of loaded tensors
+   */
   unsigned int getNumLoadedTensors();
 
 private:
diff --git a/nntrainer/tensor/tensor_pool.h b/nntrainer/tensor/tensor_pool.h
index 9b1f2c887..f5aaae324 100644
--- a/nntrainer/tensor/tensor_pool.h
+++ b/nntrainer/tensor/tensor_pool.h
@@ -304,6 +304,11 @@ class TensorPool {
    */
   void loadCacheCancel(int id);
 
+  /**
+   * @brief Get number of loaded tensors
+   *
+   * @return number of loaded tensors
+   */
   unsigned int getNumLoadedTensors();
 
 private: