DepthToSpaceEndToEndTestImpl.hpp File Reference

#include <ResolveType.hpp>
#include <QuantizeHelper.hpp>
#include <backendsCommon/test/DataLayoutUtils.hpp>

Go to the source code of this file.

Functions
template<armnn::DataType ArmnnType>
void	DepthToSpaceEndToEnd (const std::vector< armnn::BackendId > &defaultBackends, armnn::DataLayout dataLayout)

Function Documentation

DepthToSpaceEndToEnd()

void DepthToSpaceEndToEnd	(	const std::vector< armnn::BackendId > &	defaultBackends,
		armnn::DataLayout	dataLayout
	)

Definition at line 82 of file DepthToSpaceEndToEndTestImpl.hpp.

{
    using namespace armnn;

    TensorShape inputShape  = { 2, 2, 2, 4 };
    TensorShape outputShape = { 2, 4, 4, 1 };

    std::vector<float> inputData =
    {
         1.f,  2.f,  3.f,  4.f,
         5.f,  6.f,  7.f,  8.f,
         9.f, 10.f, 11.f, 12.f,
        13.f, 14.f, 15.f, 16.f,

        17.f, 18.f, 19.f, 20.f,
        21.f, 22.f, 23.f, 24.f,
        25.f, 26.f, 27.f, 28.f,
        29.f, 30.f, 31.f, 32.f
    };

    std::vector<float> expectedOutputData =
    {
         1.f,  2.f,  5.f,  6.f,
         3.f,  4.f,  7.f,  8.f,
         9.f, 10.f, 13.f, 14.f,
        11.f, 12.f, 15.f, 16.f,

        17.f, 18.f, 21.f, 22.f,
        19.f, 20.f, 23.f, 24.f,
        25.f, 26.f, 29.f, 30.f,
        27.f, 28.f, 31.f, 32.f
    };

    DepthToSpaceEndToEndImpl<ArmnnType>(defaultBackends,
                                        DepthToSpaceDescriptor(2, dataLayout),
                                        inputShape,
                                        outputShape,
                                        inputData,
                                        expectedOutputData);
}