20.11
Cifar10Database.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
7
#include "
ClassifierTestCaseData.hpp
"
8
9
#include <string>
10
#include <memory>
11
12
class
Cifar10Database
13
{
14
public
:
15
using
DataType
= float;
16
using
TTestCaseData
=
ClassifierTestCaseData<DataType>
;
17
18
explicit
Cifar10Database
(
const
std::string& binaryFileDirectory,
bool
rgbPack =
false
);
19
std::unique_ptr<TTestCaseData>
GetTestCaseData
(
unsigned
int
testCaseId);
20
21
private
:
22
std::string m_BinaryDirectory;
23
bool
m_RgbPack;
24
};
ClassifierTestCaseData
Definition:
ClassifierTestCaseData.hpp:10
Cifar10Database
Definition:
Cifar10Database.hpp:12
ClassifierTestCaseData.hpp
Cifar10Database::DataType
float DataType
Definition:
Cifar10Database.hpp:15
Cifar10Database::GetTestCaseData
std::unique_ptr< TTestCaseData > GetTestCaseData(unsigned int testCaseId)
Definition:
Cifar10Database.cpp:20
Cifar10Database::Cifar10Database
Cifar10Database(const std::string &binaryFileDirectory, bool rgbPack=false)
Definition:
Cifar10Database.cpp:15
tests
Cifar10Database.hpp
Generated on Wed Nov 25 2020 09:59:14 for ArmNN by
1.8.13