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 <backendsCommon/test/DataTypeUtils.hpp>
9 #include <backendsCommon/test/TensorCopyUtils.hpp>
10 #include <backendsCommon/test/WorkloadTestUtils.hpp>
11
12 #include <test/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 auto input = MakeTensor<int32_t, 1>(inputTensorInfo, ConvertToDataType<armnn::DataType::Signed32>(
25 {2, 2, 3, 2},
26 inputTensorInfo));
27
28 LayerTestResult<T, 4> ret(outputTensorInfo);
29 ret.outputExpected = MakeTensor<T, 4>(outputTensorInfo, ConvertToDataType<ArmnnType>(
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 1.0f, 1.0f, 1.0f, 1.0f, 1.0f, 1.0f, 1.0f, 1.0f, 1.0f, 1.0f, 1.0f, 1.0f },
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.CreateFill(data, info);
44
45 inputHandle->Allocate();
46 outputHandle->Allocate();
47
48 CopyDataToITensorHandle(inputHandle.get(), &input[0]);
49
50 workload->Execute();
51
52 CopyDataFromITensorHandle(&ret.output[0][0][0][0], outputHandle.get());
53
54 return ret;
55 }
56
57 //
58 // Explicit template specializations
59 //
60 template LayerTestResult<armnn::ResolveType<armnn::DataType::Float32>, 4>
61 SimpleFillTest<armnn::DataType::Float32>(
62 armnn::IWorkloadFactory& workloadFactory,
63 const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager,
64 const armnn::ITensorHandleFactory& tensorHandleFactory);
65
66 template LayerTestResult<armnn::ResolveType<armnn::DataType::Float16>, 4>
67 SimpleFillTest<armnn::DataType::Float16>(
68 armnn::IWorkloadFactory& workloadFactory,
69 const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager,
70 const armnn::ITensorHandleFactory& tensorHandleFactory);
71
72 template LayerTestResult<armnn::ResolveType<armnn::DataType::Signed32>, 4>
73 SimpleFillTest<armnn::DataType::Signed32>(
74 armnn::IWorkloadFactory& workloadFactory,
75 const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager,
76 const armnn::ITensorHandleFactory& tensorHandleFactory);