[Ahub] Fix Ahub issue

Previously, nntrainer only supported fp32. nntrainer didn't need to
change the data type, but it needs to be changed to support fp16. If
the tensor type is fp16, get input_dim of InitLayerContext through
getInputDimentions and call setDataType. For getInputDimentions,
return a const object. Added getMutableInputDimentions function
because return object of getInputDimentions cannot be modified.

Signed-off-by: SeoHyungjun <hyungjun.seo@samsung.com>

diff --git a/nntrainer/layers/layer_context.h b/nntrainer/layers/layer_context.h
index 35653bde0f46d5f3d35149af702994bfd2fc10f2..9fc7abff69ef73bde1925fe6a09d17ee92d0cbae 100644 (file)
--- a/nntrainer/layers/layer_context.h
+++ b/nntrainer/layers/layer_context.h
@@ -112,6 +112,13 @@ public:
    */
   const std::vector<TensorDim> &getInputDimensions() const { return input_dim; }
 
+  /**
+   * @brief Get the Mutable Input Dimensions object
+   *
+   * @return std::vector<TensorDim>& Input dimensions
+   */
+  std::vector<TensorDim> &getMutableInputDimensions() { return input_dim; }
+
   /**
    * @brief Set Data Type for Input Dimensions
    *

diff --git a/test/unittest/layers/layers_golden_tests.cpp b/test/unittest/layers/layers_golden_tests.cpp
index 4980cde10b22d03060ecb212147c6d0a80f8fe13..eac8872750509cef424df757edf8881bcda843e0 100644 (file)
--- a/test/unittest/layers/layers_golden_tests.cpp
+++ b/test/unittest/layers/layers_golden_tests.cpp
@@ -70,7 +70,7 @@ createInitContext(Layer *layer, const std::string &input_shape_str,
                            "golden_test", "", 0.0, tensor_type);
   layer->finalize(context);
 
-  for (auto dim : context.getInputDimensions()) {
+  for (auto &dim : context.getMutableInputDimensions()) {
     if (tensor_type[2] == "fp16") {
       dim.setDataType(ml::train::TensorDim::DataType::FP16);
     }