blob: bcb0d6f124754e19fd45572056b9730f04d1649a [file] [log] [blame]
Narumol Prangnawarat7ddbbae2020-03-13 10:26:05 +00001//
2// Copyright © 2020 Arm Ltd. All rights reserved.
3// SPDX-License-Identifier: MIT
4//
5
6#pragma once
7
Sadik Armagana097d2a2021-11-24 15:47:28 +00008#include <armnnTestUtils/LayerTestResult.hpp>
Narumol Prangnawarat7ddbbae2020-03-13 10:26:05 +00009
10#include <BFloat16.hpp>
11
12#include <armnn/backends/IBackendInternal.hpp>
Colm Donelan0c479742021-12-10 12:43:54 +000013#include <armnn/backends/WorkloadFactory.hpp>
Narumol Prangnawarat7ddbbae2020-03-13 10:26:05 +000014
15LayerTestResult<float, 4> ConvertBf16ToFp32Test(
16 armnn::IWorkloadFactory& workloadFactory,
Keith Davisf500d6c2020-08-31 08:32:55 +010017 const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager,
18 const armnn::ITensorHandleFactory& tensorHandleFactory);