commit d8eee59735526ead6b87343c3ed9069e682b6e8c
author Nattapat Chaimanowong <nattapat.chaimanowong@arm.com> Fri Oct 26 10:24:14 2018 +0100
committer nattapat.chaimanowong <nattapat.chaimanowong@arm.com> Fri Oct 26 12:38:34 2018 +0000
tree 44b994c98857b28674fb4911cb20489aaaecc437
parent d4dfa684941a21314b70593d01b0fc2167eebad4

IVGCVSW-2029 Fix fully connected layer support in TfLite Parser and implement test for TfLite VGG16 quantized

Change-Id: I2061f62f62684b963fa0f090718f1dcffe5c93ce

tests/ImagePreprocessor.hpp

diff --git a/tests/ImagePreprocessor.hpp b/tests/ImagePreprocessor.hpp
index 9add6d8..d77113c 100644
--- a/tests/ImagePreprocessor.hpp
+++ b/tests/ImagePreprocessor.hpp
@@ -37,10 +37,12 @@
         int32_t offset=0,
         const std::array<float, 3> mean={{0, 0, 0}},
         const std::array<float, 3> stddev={{1, 1, 1}},
-        DataFormat dataFormat=DataFormat::NHWC)
+        DataFormat dataFormat=DataFormat::NHWC,
+        unsigned int batchSize=1)
     : m_BinaryDirectory(binaryFileDirectory)
     , m_Height(height)
     , m_Width(width)
+    , m_BatchSize(batchSize)
     , m_Scale(scale)
     , m_Offset(offset)
     , m_ImageSet(imageSet)
@@ -61,6 +63,7 @@
     std::string m_BinaryDirectory;
     unsigned int m_Height;
     unsigned int m_Width;
+    unsigned int m_BatchSize;
     // Quantization parameters
     float m_Scale;
     int32_t m_Offset;