diff options
| author | Nattapat Chaimanowong <nattapat.chaimanowong@arm.com> | 2018-10-26 10:24:14 +0100 |
|---|---|---|
| committer | nattapat.chaimanowong <nattapat.chaimanowong@arm.com> | 2018-10-26 12:38:34 +0000 |
| commit | d8eee59735526ead6b87343c3ed9069e682b6e8c (patch) | |
| tree | 44b994c98857b28674fb4911cb20489aaaecc437 /tests/ImagePreprocessor.hpp | |
| parent | d4dfa684941a21314b70593d01b0fc2167eebad4 (diff) | |
| download | armnn-d8eee59735526ead6b87343c3ed9069e682b6e8c.tar.gz | |
IVGCVSW-2029 Fix fully connected layer support in TfLite Parser and implement test for TfLite VGG16 quantized
Change-Id: I2061f62f62684b963fa0f090718f1dcffe5c93ce
Diffstat (limited to 'tests/ImagePreprocessor.hpp')
| -rw-r--r-- | tests/ImagePreprocessor.hpp | 5 |
1 files changed, 4 insertions, 1 deletions
diff --git a/tests/ImagePreprocessor.hpp b/tests/ImagePreprocessor.hpp index 9add6d86ad..d77113c6d9 100644 --- a/tests/ImagePreprocessor.hpp +++ b/tests/ImagePreprocessor.hpp @@ -37,10 +37,12 @@ public: 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 @@ private: 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; |