23.02
|
Go to the documentation of this file.
29 auto convertFunc = [](uint8_t* dst,
const uint8_t* src,
size_t size)
31 auto input =
reinterpret_cast<const Half*
>(src);
32 auto output =
reinterpret_cast<float*
>(dst);
33 size_t numElements = size/2;
37 for (
const auto& pair : m_TensorHandlePairs)
76 void NeonConvertFp16ToFp32Workload::Reconfigure()
arm::pipe::ProfilingGuid GetGuid() const final
void ValidateInputsOutputs(const std::string &descName, unsigned int numExpectedIn, unsigned int numExpectedOut) const
Copyright (c) 2021 ARM Limited and Contributors.
void GatherTensorHandlePairs(const DescriptorType &descriptor, std::vector< std::pair< SrcTensorHandleType *, DstTensorHandleType * >> &tensorHandlePairs)
#define ARMNN_SCOPED_PROFILING_EVENT_NEON_GUID(name, guid)
void ReplaceOutputTensorHandle(ITensorHandle *tensorHandle, unsigned int slot) override
virtual void Execute() const override
NeonConvertFp16ToFp32Workload(const ConvertFp16ToFp32QueueDescriptor &descriptor, const WorkloadInfo &info)
Contains information about TensorInfos of a layer.
void ReplaceInputTensorHandle(ITensorHandle *tensorHandle, unsigned int slot) override
void CopyTensorContentsGeneric(const ITensorHandle *srcTensor, ITensorHandle *dstTensor, CopyFunc copy)
static void ConvertFloat16To32(const void *srcFloat16Buffer, size_t numElements, float *dstFloat32Buffer)
std::vector< ITensorHandle * > m_Inputs