xref: /aosp_15_r20/external/armnn/tests/ClassifierTestCaseData.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 <vector>
8*89c4ff92SAndroid Build Coastguard Worker 
9*89c4ff92SAndroid Build Coastguard Worker template <typename DataType>
10*89c4ff92SAndroid Build Coastguard Worker class ClassifierTestCaseData
11*89c4ff92SAndroid Build Coastguard Worker {
12*89c4ff92SAndroid Build Coastguard Worker public:
ClassifierTestCaseData(unsigned int label,std::vector<DataType> inputImage)13*89c4ff92SAndroid Build Coastguard Worker     ClassifierTestCaseData(unsigned int label, std::vector<DataType> inputImage)
14*89c4ff92SAndroid Build Coastguard Worker      : m_Label(label)
15*89c4ff92SAndroid Build Coastguard Worker      , m_InputImage(std::move(inputImage))
16*89c4ff92SAndroid Build Coastguard Worker     {
17*89c4ff92SAndroid Build Coastguard Worker     }
18*89c4ff92SAndroid Build Coastguard Worker 
19*89c4ff92SAndroid Build Coastguard Worker     const unsigned int m_Label;
20*89c4ff92SAndroid Build Coastguard Worker     std::vector<DataType> m_InputImage;
21*89c4ff92SAndroid Build Coastguard Worker };
22