xref: /aosp_15_r20/external/armnn/tests/Cifar10Database.hpp (revision 89c4ff92f2867872bb9e2354d150bf0c8c502810) 
1*89c4ff92SAndroid Build Coastguard Worker //
2*89c4ff92SAndroid Build Coastguard Worker // Copyright © 2017 Arm Ltd. All rights reserved.
3*89c4ff92SAndroid Build Coastguard Worker // SPDX-License-Identifier: MIT
4*89c4ff92SAndroid Build Coastguard Worker //
5*89c4ff92SAndroid Build Coastguard Worker #pragma once
6*89c4ff92SAndroid Build Coastguard Worker 
7*89c4ff92SAndroid Build Coastguard Worker #include "ClassifierTestCaseData.hpp"
8*89c4ff92SAndroid Build Coastguard Worker 
9*89c4ff92SAndroid Build Coastguard Worker #include <string>
10*89c4ff92SAndroid Build Coastguard Worker #include <memory>
11*89c4ff92SAndroid Build Coastguard Worker 
12*89c4ff92SAndroid Build Coastguard Worker class Cifar10Database
13*89c4ff92SAndroid Build Coastguard Worker {
14*89c4ff92SAndroid Build Coastguard Worker public:
15*89c4ff92SAndroid Build Coastguard Worker     using DataType = float;
16*89c4ff92SAndroid Build Coastguard Worker     using TTestCaseData = ClassifierTestCaseData<DataType>;
17*89c4ff92SAndroid Build Coastguard Worker 
18*89c4ff92SAndroid Build Coastguard Worker     explicit Cifar10Database(const std::string& binaryFileDirectory, bool rgbPack = false);
19*89c4ff92SAndroid Build Coastguard Worker     std::unique_ptr<TTestCaseData> GetTestCaseData(unsigned int testCaseId);
20*89c4ff92SAndroid Build Coastguard Worker 
21*89c4ff92SAndroid Build Coastguard Worker private:
22*89c4ff92SAndroid Build Coastguard Worker     std::string m_BinaryDirectory;
23*89c4ff92SAndroid Build Coastguard Worker     bool m_RgbPack;
24*89c4ff92SAndroid Build Coastguard Worker };
25