ResolveType.hpp

Go to the documentation of this file.

//
// Copyright © 2017 Arm Ltd and Contributors. All rights reserved.
// SPDX-License-Identifier: MIT
//
 
#pragma once
 
#include "armnn/Types.hpp"
#include "BFloat16.hpp"
#include "Half.hpp"
 
namespace armnn
{
 
template<DataType DT>
struct ResolveTypeImpl;
 
template <>
struct ResolveTypeImpl<DataType::Float16>
{
    using Type = Half;
};
 
template<>
struct ResolveTypeImpl<DataType::Float32>
{
    using Type = float;
};
 
template<>
struct ResolveTypeImpl<DataType::QAsymmU8>
{
    using Type = uint8_t;
};
 
template<>
struct ResolveTypeImpl<DataType::QAsymmS8>
{
    using Type = int8_t;
};
 
template<>
struct ResolveTypeImpl<DataType::QSymmS8>
{
    using Type = int8_t;
};
 
template<>
struct ResolveTypeImpl<DataType::QSymmS16>
{
    using Type = int16_t;
};
 
template<>
struct ResolveTypeImpl<DataType::Signed32>
{
    using Type = int32_t;
};
 
template<>
struct ResolveTypeImpl<DataType::Signed64>
{
    using Type = int64_t;
};
 
template<>
struct ResolveTypeImpl<DataType::Boolean>
{
    using Type = uint8_t;
};
 
template<>
struct ResolveTypeImpl<DataType::BFloat16>
{
    using Type = BFloat16;
};
 
template<DataType DT>
using ResolveType = typename ResolveTypeImpl<DT>::Type;
 
} //namespace armnn