ArmNN
 20.02
MnistDatabase.hpp
Go to the documentation of this file.
1 //
2 // Copyright © 2017 Arm Ltd. All rights reserved.
3 // SPDX-License-Identifier: MIT
4 //
5 #pragma once
6 
8 
9 #include <string>
10 #include <memory>
11 
13 {
14 public:
15  using DataType = float;
17 
18  explicit MnistDatabase(const std::string& binaryFileDirectory, bool scaleValues = false);
19  std::unique_ptr<TTestCaseData> GetTestCaseData(unsigned int testCaseId);
20 
21 private:
22  std::string m_BinaryDirectory;
23  bool m_ScaleValues;
24 };
std::unique_ptr< TTestCaseData > GetTestCaseData(unsigned int testCaseId)
MnistDatabase(const std::string &binaryFileDirectory, bool scaleValues=false)