1 // 2 // Copyright © 2020 Arm Ltd. All rights reserved. 3 // SPDX-License-Identifier: MIT 4 // 5 6 #pragma once 7 8 #include "LayerTestResult.hpp" 9 10 #include <BFloat16.hpp> 11 12 #include <armnn/backends/IBackendInternal.hpp> 13 #include <backendsCommon/WorkloadFactory.hpp> 14 15 LayerTestResult<armnn::BFloat16, 4> ConvertFp32ToBf16Test( 16 armnn::IWorkloadFactory& workloadFactory, 17 const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager, 18 const armnn::ITensorHandleFactory& tensorHandleFactory); 19