Fix build break (#501)

This commit fixes build breaks.

Signed-off-by: Jonghyun Park <jh1302.park@samsung.com>

diff --git a/contrib/nnkit/contrib/nnsuite/conv/tflite/ConvBackend.test.cpp b/contrib/nnkit/contrib/nnsuite/conv/tflite/ConvBackend.test.cpp
index f7be071..07b4dce 100644 (file)
--- a/contrib/nnkit/contrib/nnsuite/conv/tflite/ConvBackend.test.cpp
+++ b/contrib/nnkit/contrib/nnsuite/conv/tflite/ConvBackend.test.cpp
@@ -52,7 +52,7 @@ private:
   const feature::Shape _ofm_shape;
 
 private:
-  const kernel::Overlay<float> _ker;
+  const kernel::Overlay<float, float *> _ker;
 };
 
 TEST(CONV_BACKEND, conv_3x3)

diff --git a/contrib/nnkit/contrib/nnsuite/conv/tflite/Entry.cpp b/contrib/nnkit/contrib/nnsuite/conv/tflite/Entry.cpp
index f979a7c..cf20878 100644 (file)
--- a/contrib/nnkit/contrib/nnsuite/conv/tflite/Entry.cpp
+++ b/contrib/nnkit/contrib/nnsuite/conv/tflite/Entry.cpp
@@ -1,6 +1,6 @@
 #include "ConvBackend.h"
 
-#include <nnsuite/conv/RandomFixedModel.h>
+#include <nnsuite/conv/RandomModel.h>
 
 #include <nnkit/Backend.h>
 #include <nnkit/CmdlineArguments.h>
@@ -22,7 +22,7 @@ extern "C" std::unique_ptr<nnkit::Backend> make_backend(const nnkit::CmdlineArgu
 
   std::cout << "SEED: " << seed << std::endl;
 
-  const nnsuite::conv::RandomFixedModel model{seed};
+  const nnsuite::conv::RandomModel model{seed};
 
   return nncc::foundation::make_unique<ConvBackend>(model);
 }