1 //
2 // Copyright © 2020 Arm Ltd and Contributors. All rights reserved.
3 // SPDX-License-Identifier: MIT
4 //
5
6 #include "FillTestImpl.hpp"
7
8 #include <DataTypeUtils.hpp>
9 #include <armnnTestUtils/TensorCopyUtils.hpp>
10 #include <armnnTestUtils/WorkloadTestUtils.hpp>
11
12 #include <armnnTestUtils/TensorHelpers.hpp>
13
14 template<armnn::DataType ArmnnType, typename T>
SimpleFillTest(armnn::IWorkloadFactory & workloadFactory,const armnn::IBackendInternal::IMemoryManagerSharedPtr & memoryManager,const armnn::ITensorHandleFactory & tensorHandleFactory)15 LayerTestResult<T, 4> SimpleFillTest(
16 armnn::IWorkloadFactory& workloadFactory,
17 const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager,
18 const armnn::ITensorHandleFactory& tensorHandleFactory)
19 {
20 IgnoreUnused(memoryManager);
21 armnn::TensorInfo inputTensorInfo({4}, armnn::DataType::Signed32);
22 armnn::TensorInfo outputTensorInfo({2, 2, 3, 2}, ArmnnType);
23
24 std::vector<int32_t> input = ConvertToDataType<armnn::DataType::Signed32>( { 2, 2, 3, 2 }, inputTensorInfo);
25
26 std::vector<T> actualOutput(outputTensorInfo.GetNumElements());
27 std::vector<T> expectedOutput = ConvertToDataType<ArmnnType>(
28 {
29 1.0f, 1.0f, 1.0f, 1.0f, 1.0f, 1.0f, 1.0f, 1.0f, 1.0f, 1.0f, 1.0f, 1.0f,
30 1.0f, 1.0f, 1.0f, 1.0f, 1.0f, 1.0f, 1.0f, 1.0f, 1.0f, 1.0f, 1.0f, 1.0f
31 },
32 outputTensorInfo);
33
34 std::unique_ptr<armnn::ITensorHandle> inputHandle = tensorHandleFactory.CreateTensorHandle(inputTensorInfo);
35 std::unique_ptr<armnn::ITensorHandle> outputHandle = tensorHandleFactory.CreateTensorHandle(outputTensorInfo);
36
37 armnn::FillQueueDescriptor data;
38 data.m_Parameters.m_Value = 1.0f;
39 armnn::WorkloadInfo info;
40 AddInputToWorkload(data, info, inputTensorInfo, inputHandle.get());
41 AddOutputToWorkload(data, info, outputTensorInfo, outputHandle.get());
42
43 std::unique_ptr<armnn::IWorkload> workload = workloadFactory.CreateWorkload(armnn::LayerType::Fill, data, info);
44
45 inputHandle->Allocate();
46 outputHandle->Allocate();
47
48 CopyDataToITensorHandle(inputHandle.get(), input.data());
49
50 workload->Execute();
51
52 CopyDataFromITensorHandle(actualOutput.data(), outputHandle.get());
53
54 return LayerTestResult<T, 4>(actualOutput,
55 expectedOutput,
56 outputHandle->GetShape(),
57 outputTensorInfo.GetShape());
58 }
59
60 //
61 // Explicit template specializations
62 //
63 template LayerTestResult<armnn::ResolveType<armnn::DataType::Float32>, 4>
64 SimpleFillTest<armnn::DataType::Float32>(
65 armnn::IWorkloadFactory& workloadFactory,
66 const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager,
67 const armnn::ITensorHandleFactory& tensorHandleFactory);
68
69 template LayerTestResult<armnn::ResolveType<armnn::DataType::Float16>, 4>
70 SimpleFillTest<armnn::DataType::Float16>(
71 armnn::IWorkloadFactory& workloadFactory,
72 const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager,
73 const armnn::ITensorHandleFactory& tensorHandleFactory);
74
75 template LayerTestResult<armnn::ResolveType<armnn::DataType::Signed32>, 4>
76 SimpleFillTest<armnn::DataType::Signed32>(
77 armnn::IWorkloadFactory& workloadFactory,
78 const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager,
79 const armnn::ITensorHandleFactory& tensorHandleFactory);