#include <FloatingPointConverter.hpp>
|
| static void | ConvertFloat32To16 (const float *srcFloat32Buffer, size_t numElements, void *dstFloat16Buffer) |
| | Converts a buffer of FP32 values to FP16, and stores in the given dstFloat16Buffer.
|
| static void | ConvertFloat16To32 (const void *srcFloat16Buffer, size_t numElements, float *dstFloat32Buffer) |
| static void | ConvertFloat32ToBFloat16 (const float *srcFloat32Buffer, size_t numElements, void *dstBFloat16Buffer) |
| static void | ConvertBFloat16ToFloat32 (const void *srcBFloat16Buffer, size_t numElements, float *dstFloat32Buffer) |
Definition at line 13 of file FloatingPointConverter.hpp.
◆ ConvertBFloat16ToFloat32()
| void ConvertBFloat16ToFloat32 |
( |
const void * | srcBFloat16Buffer, |
|
|
size_t | numElements, |
|
|
float * | dstFloat32Buffer ) |
|
static |
◆ ConvertFloat16To32()
| void ConvertFloat16To32 |
( |
const void * | srcFloat16Buffer, |
|
|
size_t | numElements, |
|
|
float * | dstFloat32Buffer ) |
|
static |
◆ ConvertFloat32To16()
| void ConvertFloat32To16 |
( |
const float * | srcFloat32Buffer, |
|
|
size_t | numElements, |
|
|
void * | dstFloat16Buffer ) |
|
static |
Converts a buffer of FP32 values to FP16, and stores in the given dstFloat16Buffer.
dstFloat16Buffer should be (numElements * 2) in size
Definition at line 17 of file FloatingPointConverter.cpp.
20{
21 if (srcFloat32Buffer == nullptr)
22 {
23 throw armnn::InvalidArgumentException("ConvertFloat32To16: source float32 buffer pointer is null");
24 }
25 if (dstFloat16Buffer == nullptr)
26 {
27 throw armnn::InvalidArgumentException("ConvertFloat32To16: destination float16 buffer pointer is null");
28 }
29
31
32 for (size_t i = 0; i < numElements; i++)
33 {
35 if (isinf(pHalf[i]))
36 {
37
38 pHalf[i] = copysign(std::numeric_limits<armnn::Half>::max(), pHalf[i]);
39 }
40 }
41}
Referenced by NeonConvertFp32ToFp16Workload::Execute(), Float32ToFloat16::Func(), and Float16Encoder::Set().
◆ ConvertFloat32ToBFloat16()
| void ConvertFloat32ToBFloat16 |
( |
const float * | srcFloat32Buffer, |
|
|
size_t | numElements, |
|
|
void * | dstBFloat16Buffer ) |
|
static |
The documentation for this class was generated from the following files: