21.05
|
#include "ConvertFp32ToBf16TestImpl.hpp"
#include <backendsCommon/test/TensorCopyUtils.hpp>
#include <backendsCommon/test/WorkloadTestUtils.hpp>
#include <test/TensorHelpers.hpp>
Go to the source code of this file.
Functions | |
LayerTestResult< armnn::BFloat16, 4 > | ConvertFp32ToBf16Test (armnn::IWorkloadFactory &workloadFactory, const armnn::IBackendInternal::IMemoryManagerSharedPtr &memoryManager, const armnn::ITensorHandleFactory &tensorHandleFactory) |
LayerTestResult<armnn::BFloat16, 4> ConvertFp32ToBf16Test | ( | armnn::IWorkloadFactory & | workloadFactory, |
const armnn::IBackendInternal::IMemoryManagerSharedPtr & | memoryManager, | ||
const armnn::ITensorHandleFactory & | tensorHandleFactory | ||
) |
Definition at line 13 of file ConvertFp32ToBf16TestImpl.cpp.
References armnn::BFloat16, CopyDataFromITensorHandle(), CopyDataToITensorHandle(), IWorkloadFactory::CreateConvertFp32ToBf16(), ITensorHandleFactory::CreateTensorHandle(), armnn::Float32, and armnn::IgnoreUnused().
Referenced by BOOST_AUTO_TEST_CASE().