21 inline std::unique_ptr<Decoder<float>> MakeSigned32PerAxisDecoder(
const TensorInfo& info,
const void* data)
23 return std::make_unique<ScaledInt32PerAxisDecoder>(
static_cast<const int32_t*
>(data), info);
26 inline std::unique_ptr<Decoder<float>> MakeSigned32Decoder(
const TensorInfo& info,
const void* data)
28 if(info.HasMultipleQuantizationScales())
32 return MakeSigned32PerAxisDecoder(info, data);
36 if (info.GetQuantizationDim().has_value())
41 return MakeSigned32PerAxisDecoder(info, data);
44 const float scale = info.GetQuantizationScale();
50 return std::make_unique<Int32Decoder>(
static_cast<const int32_t*
>(data));
56 return std::make_unique<ScaledInt32Decoder>(
static_cast<const int32_t*
>(data), scale);
63 inline std::unique_ptr<Decoder<T>>
MakeDecoder(
const TensorInfo& info,
const void* data =
nullptr);
72 return std::make_unique<QASymmS8Decoder>(
73 static_cast<const int8_t*
>(data),
79 return std::make_unique<QASymm8Decoder>(
80 static_cast<const uint8_t*
>(data),
86 return std::make_unique<QSymm16Decoder>(
87 static_cast<const int16_t*
>(data),
93 return std::make_unique<BFloat16Decoder>(
static_cast<const BFloat16*
>(data));
97 return std::make_unique<Float16Decoder>(
static_cast<const Half*
>(data));
101 return std::make_unique<Float32Decoder>(
static_cast<const float*
>(data));
105 return MakeSigned32Decoder(info, data);
112 return std::make_unique<QSymm8PerAxisDecoder>(
static_cast<const int8_t*
>(data), info);
116 return std::make_unique<QSymmS8Decoder>(
117 static_cast<const int8_t*
>(data),
124 return std::make_unique<BooleanDecoder>(
static_cast<const uint8_t*
>(data));
142 return std::make_unique<BooleanDecoderBool>(
static_cast<const uint8_t*
>(data));
160 return std::make_unique<Int32ToInt32tDecoder>(
static_cast<const int32_t*
>(data));
bool HasPerAxisQuantization() const
std::pair< unsigned int, std::vector< float > > GetPerAxisParams(const armnn::TensorInfo &info)
Copyright (c) 2021 ARM Limited and Contributors.
std::unique_ptr< Decoder< T > > MakeDecoder(const TensorInfo &info, const void *data=nullptr)
#define ARMNN_ASSERT_MSG(COND, MSG)
int32_t GetQuantizationOffset() const
float GetQuantizationScale() const
DataType GetDataType() const