aboutsummaryrefslogtreecommitdiff
path: root/include/armnnUtils/TContainer.hpp
blob: d2a868ac182c7c73843bb589bf2efa3a5a446c80 (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
//
// Copyright © 2020 Arm Ltd. All rights reserved.
// SPDX-License-Identifier: MIT
//

#pragma once

#include <armnn/TypesUtils.hpp>
#include <Half.hpp>

#include <mapbox/variant.hpp>

namespace armnnUtils
{

// Standard declaration of TContainer used by ArmNN
// Changes to this declaration constitute an api/abi break, new types should be added as a separate declaration and
// merged on the next planned api/abi update.
using TContainer = mapbox::util::variant<std::vector<float>,
                                         std::vector<int>,
                                         std::vector<uint8_t>,
                                         std::vector<int8_t>,
                                         std::vector<armnn::Half>>;

} // namespace armnnUtils