FakeQuantizationTestImpl.hpp File Reference

#include <armnnTestUtils/LayerTestResult.hpp>
#include <armnn/backends/IBackendInternal.hpp>
#include <armnn/backends/WorkloadFactory.hpp>

Go to the source code of this file.

Functions
LayerTestResult< float, 2 >	FakeQuantizationTest (armnn::IWorkloadFactory &workloadFactory, const armnn::IBackendInternal::IMemoryManagerSharedPtr &memoryManager, const armnn::ITensorHandleFactory &tensorHandleFactory)

Function Documentation

FakeQuantizationTest()

LayerTestResult<float, 2> FakeQuantizationTest	(	armnn::IWorkloadFactory &	workloadFactory,
		const armnn::IBackendInternal::IMemoryManagerSharedPtr &	memoryManager,
		const armnn::ITensorHandleFactory &	tensorHandleFactory
	)

Definition at line 16 of file FakeQuantizationTestImpl.cpp.

References CopyDataFromITensorHandle(), CopyDataToITensorHandle(), ITensorHandleFactory::CreateTensorHandle(), IWorkloadFactory::CreateWorkload(), armnn::FakeQuantization, armnn::Float32, armnn::IgnoreUnused(), FakeQuantizationDescriptor::m_Max, FakeQuantizationDescriptor::m_Min, and QueueDescriptorWithParameters< LayerDescriptor >::m_Parameters.

Referenced by TEST_SUITE().

{
    IgnoreUnused(memoryManager);
    constexpr unsigned int width = 2;
    constexpr unsigned int height = 3;

    const armnn::TensorInfo tensorInfo({ height, width }, armnn::DataType::Float32);

    std::vector<float> input =
    {
       -10.0f, -5.0f,
         0.0f,  5.0f,
        10.0f, 10.0f
    };

    std::vector<float> actualOutput(tensorInfo.GetNumElements());
    std::vector<float> expectedOutput(tensorInfo.GetNumElements());

    std::unique_ptr<armnn::ITensorHandle> inputHandle   = tensorHandleFactory.CreateTensorHandle(tensorInfo);
    std::unique_ptr<armnn::ITensorHandle> outputHandle  = tensorHandleFactory.CreateTensorHandle(tensorInfo);

    armnn::FakeQuantizationQueueDescriptor data;
    armnn::WorkloadInfo info;

    AddInputToWorkload(data, info, tensorInfo, inputHandle.get());
    AddOutputToWorkload(data, info, tensorInfo, outputHandle.get());

    float min = -10.f;
    float max =  10.f;

    data.m_Parameters.m_Min = min;
    data.m_Parameters.m_Max = max;

    armnn::PassthroughTensorHandle refHandle(tensorInfo, expectedOutput.data());
    armnn::FakeQuantizationQueueDescriptor refData = data;
    armnn::WorkloadInfo refInfo = info;
    SetWorkloadOutput(refData, refInfo, 0, tensorInfo, &refHandle);

    std::unique_ptr<armnn::IWorkload> workload = workloadFactory.CreateWorkload(armnn::LayerType::FakeQuantization,
                                                                                data,
                                                                                info);

    inputHandle->Allocate();
    outputHandle->Allocate();

    CopyDataToITensorHandle(inputHandle.get(), input.data());

    workload->PostAllocationConfigure();
    workload->Execute();

    CopyDataFromITensorHandle(actualOutput.data(), outputHandle.get());

    expectedOutput =
    {
        0.0f,     63.0f,
        128.0f,   191.0f,
        255.0f,   255.0f
    };

    return LayerTestResult<float, 2>(actualOutput,
                                     expectedOutput,
                                     outputHandle->GetShape(),
                                     tensorInfo.GetShape());
}