23 template<armnn::DataType ArmnnType,
typename T = armnn::ResolveType<ArmnnType>>
35 if (armnn::IsQuantizedType<T>())
38 inputTensorInfo.SetQuantizationOffset(128);
39 alphaTensorInfo.SetQuantizationScale(0.25f);
40 alphaTensorInfo.SetQuantizationOffset(50);
41 outputTensorInfo.SetQuantizationScale(0.5f);
42 outputTensorInfo.SetQuantizationOffset(120);
45 std::vector<float> inputData
49 0.0f, 0.0f, 0.0f, 1.0f, 1.0f, 1.0f, -1.0f, -1.0f, -1.0f, -2.0f, -2.0f, -2.0f
51 std::vector<float> alphaData
57 std::vector<float> outputExpectedData =
61 0.0f, 0.0f, 0.0f, 1.0f, 1.0f, 1.0f, 0.0f, -1.0f, -2.0f, 0.0f, -2.0f, -4.0f
64 std::vector<T> input = armnnUtils::QuantizedVector<T>(inputData,
65 inputTensorInfo.GetQuantizationScale(),
66 inputTensorInfo.GetQuantizationOffset());
68 std::vector<T> alpha = armnnUtils::QuantizedVector<T>(alphaData,
69 alphaTensorInfo.GetQuantizationScale(),
70 alphaTensorInfo.GetQuantizationOffset());
72 std::vector<T> actualOutput(outputTensorInfo.GetNumElements());
73 std::vector<T> expectedOutput = armnnUtils::QuantizedVector<T>(outputExpectedData,
74 outputTensorInfo.GetQuantizationScale(),
75 outputTensorInfo.GetQuantizationOffset());
77 std::unique_ptr <armnn::ITensorHandle> inputHandle = tensorHandleFactory.
CreateTensorHandle(inputTensorInfo);
78 std::unique_ptr <armnn::ITensorHandle> alphaHandle = tensorHandleFactory.
CreateTensorHandle(alphaTensorInfo);
79 std::unique_ptr <armnn::ITensorHandle> outputHandle = tensorHandleFactory.
CreateTensorHandle(outputTensorInfo);
83 AddInputToWorkload (descriptor, info, inputTensorInfo, inputHandle.get());
84 AddInputToWorkload (descriptor, info, alphaTensorInfo, alphaHandle.get());
85 AddOutputToWorkload(descriptor, info, outputTensorInfo, outputHandle.get());
91 inputHandle->Allocate();
92 alphaHandle->Allocate();
93 outputHandle->Allocate();
104 outputHandle->GetShape(),
105 outputTensorInfo.GetShape());
LayerTestResult< T, 4 > PreluTest(armnn::IWorkloadFactory &workloadFactory, const armnn::IBackendInternal::IMemoryManagerSharedPtr &memoryManager, const armnn::ITensorHandleFactory &tensorHandleFactory)
void IgnoreUnused(Ts &&...)
std::shared_ptr< IMemoryManager > IMemoryManagerSharedPtr
void CopyDataFromITensorHandle(void *mem, const armnn::ITensorHandle *tensorHandle)
void SetQuantizationScale(float scale)
void CopyDataToITensorHandle(armnn::ITensorHandle *tensorHandle, const void *memory)
Contains information about TensorInfos of a layer.
virtual std::unique_ptr< IWorkload > CreateWorkload(LayerType type, const QueueDescriptor &descriptor, const WorkloadInfo &info) const
virtual std::unique_ptr< ITensorHandle > CreateTensorHandle(const TensorInfo &tensorInfo) const =0