blob: e71415b6fa11bcfd4bf96a05314e84819958b47c (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
|
//
// Copyright © 2019 Arm Ltd. All rights reserved.
// SPDX-License-Identifier: MIT
//
#include <armnnUtils/QuantizeHelper.hpp>
#include <armnn/utility/IgnoreUnused.hpp>
#include <doctest/doctest.h>
#include <vector>
TEST_SUITE("QuantizeHelper")
{
namespace
{
template<typename T>
bool IsFloatIterFunc(T iter)
{
armnn::IgnoreUnused(iter);
return armnnUtils::IsFloatingPointIterator<T>::value;
}
} // anonymous namespace
TEST_CASE("IsFloatIterFuncTest")
{
std::vector<float> fArray;
CHECK(IsFloatIterFunc(fArray.begin()) == true);
CHECK(IsFloatIterFunc(fArray.cbegin()) == true);
std::vector<double> dArray;
CHECK(IsFloatIterFunc(dArray.begin()) == true);
std::vector<int> iArray;
CHECK(IsFloatIterFunc(iArray.begin()) == false);
float floats[5];
CHECK(IsFloatIterFunc(&floats[0]) == true);
int ints[5];
CHECK(IsFloatIterFunc(&ints[0]) == false);
}
}
|
