TensorTest.cpp File Reference

#include <boost/test/unit_test.hpp>
#include <armnn/Tensor.hpp>

Go to the source code of this file.

Namespaces
	armnn
	Copyright (c) 2020 ARM Limited.

Functions
std::ostream &	boost_test_print_type (std::ostream &ostr, const TensorInfo &right)
std::ostream &	boost_test_print_type (std::ostream &ostr, const TensorShape &shape)
	BOOST_FIXTURE_TEST_CASE (ConstructShapeUsingListInitialization, TensorInfoFixture)
	BOOST_FIXTURE_TEST_CASE (ConstructTensorInfo, TensorInfoFixture)
	BOOST_FIXTURE_TEST_CASE (CopyConstructTensorInfo, TensorInfoFixture)
	BOOST_FIXTURE_TEST_CASE (TensorInfoEquality, TensorInfoFixture)
	BOOST_FIXTURE_TEST_CASE (TensorInfoInequality, TensorInfoFixture)
	BOOST_FIXTURE_TEST_CASE (TensorInfoAssignmentOperator, TensorInfoFixture)
void	CheckTensor (const ConstTensor &t)
	BOOST_AUTO_TEST_CASE (TensorVsConstTensor)
	BOOST_AUTO_TEST_CASE (ModifyTensorInfo)
	BOOST_AUTO_TEST_CASE (TensorShapeOperatorBrackets)
	BOOST_AUTO_TEST_CASE (TensorInfoPerAxisQuantization)

Function Documentation

BOOST_AUTO_TEST_CASE() [1/4]

BOOST_AUTO_TEST_CASE

(

TensorVsConstTensor

)

Definition at line 105 of file TensorTest.cpp.

References CheckTensor().

{
    int mutableDatum = 2;
    const int immutableDatum = 3;

    armnn::Tensor uninitializedTensor;
    armnn::ConstTensor uninitializedTensor2;

    uninitializedTensor2 = uninitializedTensor;

    armnn::Tensor t(TensorInfo(), &mutableDatum);
    armnn::ConstTensor ct(TensorInfo(), &immutableDatum);

    // Checks that both Tensor and ConstTensor can be passed as a ConstTensor.
    CheckTensor(t);
    CheckTensor(ct);
}

BOOST_AUTO_TEST_CASE() [2/4]

BOOST_AUTO_TEST_CASE

(

ModifyTensorInfo

)

Definition at line 123 of file TensorTest.cpp.

References TensorInfo::GetDataType(), TensorInfo::GetQuantizationOffset(), TensorInfo::GetQuantizationScale(), TensorInfo::GetShape(), armnn::info, armnn::QAsymmU8, TensorInfo::SetDataType(), TensorInfo::SetQuantizationOffset(), TensorInfo::SetQuantizationScale(), and TensorInfo::SetShape().

{
    TensorInfo info;
    info.SetShape({ 5, 6, 7, 8 });
    BOOST_TEST((info.GetShape() == TensorShape({ 5, 6, 7, 8 })));
    info.SetDataType(DataType::QAsymmU8);
    BOOST_TEST((info.GetDataType() == DataType::QAsymmU8));
    info.SetQuantizationScale(10.0f);
    BOOST_TEST(info.GetQuantizationScale() == 10.0f);
    info.SetQuantizationOffset(5);
    BOOST_TEST(info.GetQuantizationOffset() == 5);
}

BOOST_AUTO_TEST_CASE() [3/4]

BOOST_AUTO_TEST_CASE

(

TensorShapeOperatorBrackets

)

Definition at line 136 of file TensorTest.cpp.

{
    TensorShape shape({0,1,2,3});
    // Checks version of operator[] which returns an unsigned int.
    BOOST_TEST(shape[2] == 2);
    // Checks the version of operator[] which returns a reference.
    shape[2] = 20;
    BOOST_TEST(shape[2] == 20);
}

BOOST_AUTO_TEST_CASE() [4/4]

BOOST_AUTO_TEST_CASE

(

TensorInfoPerAxisQuantization

)

Definition at line 146 of file TensorTest.cpp.

References BOOST_AUTO_TEST_SUITE_END(), BOOST_CHECK(), and armnn::Float32.

{
    // Old constructor
    TensorInfo tensorInfo0({ 1, 1 }, DataType::Float32, 2.0f, 1);
    BOOST_CHECK(!tensorInfo0.HasMultipleQuantizationScales());
    BOOST_CHECK(tensorInfo0.GetQuantizationScale() == 2.0f);
    BOOST_CHECK(tensorInfo0.GetQuantizationOffset() == 1);
    BOOST_CHECK(tensorInfo0.GetQuantizationScales()[0] == 2.0f);
    BOOST_CHECK(!tensorInfo0.GetQuantizationDim().has_value());

    // Set per-axis quantization scales
    std::vector<float> perAxisScales{ 3.0f, 4.0f };
    tensorInfo0.SetQuantizationScales(perAxisScales);
    BOOST_CHECK(tensorInfo0.HasMultipleQuantizationScales());
    BOOST_CHECK(tensorInfo0.GetQuantizationScales() == perAxisScales);

    // Set per-tensor quantization scale
    tensorInfo0.SetQuantizationScale(5.0f);
    BOOST_CHECK(!tensorInfo0.HasMultipleQuantizationScales());
    BOOST_CHECK(tensorInfo0.GetQuantizationScales()[0] == 5.0f);

    // Set quantization offset
    tensorInfo0.SetQuantizationDim(Optional<unsigned int>(1));
    BOOST_CHECK(tensorInfo0.GetQuantizationDim().value() == 1);

    // New constructor
    perAxisScales = { 6.0f, 7.0f };
    TensorInfo tensorInfo1({ 1, 1 }, DataType::Float32, perAxisScales, 1);
    BOOST_CHECK(tensorInfo1.HasMultipleQuantizationScales());
    BOOST_CHECK(tensorInfo1.GetQuantizationOffset() == 0);
    BOOST_CHECK(tensorInfo1.GetQuantizationScales() == perAxisScales);
    BOOST_CHECK(tensorInfo1.GetQuantizationDim().value() == 1);
}

BOOST_FIXTURE_TEST_CASE() [1/6]

BOOST_FIXTURE_TEST_CASE	(	ConstructShapeUsingListInitialization	,
		TensorInfoFixture
	)

Definition at line 53 of file TensorTest.cpp.

{
    TensorShape listInitializedShape{ 6, 7, 8, 9 };
    BOOST_TEST(listInitializedShape == m_TensorInfo.GetShape());
}

BOOST_FIXTURE_TEST_CASE() [2/6]

BOOST_FIXTURE_TEST_CASE	(	ConstructTensorInfo	,
		TensorInfoFixture
	)

Definition at line 59 of file TensorTest.cpp.

{
    BOOST_TEST(m_TensorInfo.GetNumDimensions() == 4);
    BOOST_TEST(m_TensorInfo.GetShape()[0] == 6); // <= Outer most
    BOOST_TEST(m_TensorInfo.GetShape()[1] == 7);
    BOOST_TEST(m_TensorInfo.GetShape()[2] == 8);
    BOOST_TEST(m_TensorInfo.GetShape()[3] == 9);     // <= Inner most
}

BOOST_FIXTURE_TEST_CASE() [3/6]

BOOST_FIXTURE_TEST_CASE	(	CopyConstructTensorInfo	,
		TensorInfoFixture
	)

Definition at line 68 of file TensorTest.cpp.

{
    TensorInfo copyConstructed(m_TensorInfo);
    BOOST_TEST(copyConstructed.GetNumDimensions() == 4);
    BOOST_TEST(copyConstructed.GetShape()[0] == 6);
    BOOST_TEST(copyConstructed.GetShape()[1] == 7);
    BOOST_TEST(copyConstructed.GetShape()[2] == 8);
    BOOST_TEST(copyConstructed.GetShape()[3] == 9);
}

BOOST_FIXTURE_TEST_CASE() [4/6]

BOOST_FIXTURE_TEST_CASE	(	TensorInfoEquality	,
		TensorInfoFixture
	)

Definition at line 78 of file TensorTest.cpp.

{
    TensorInfo copyConstructed(m_TensorInfo);
    BOOST_TEST(copyConstructed == m_TensorInfo);
}

BOOST_FIXTURE_TEST_CASE() [5/6]

BOOST_FIXTURE_TEST_CASE	(	TensorInfoInequality	,
		TensorInfoFixture
	)

Definition at line 84 of file TensorTest.cpp.

References armnn::Float32.

{
    TensorInfo other;
    unsigned int sizes[] = {2,3,4,5};
    other = TensorInfo(4, sizes, DataType::Float32);

    BOOST_TEST(other != m_TensorInfo);
}

BOOST_FIXTURE_TEST_CASE() [6/6]

BOOST_FIXTURE_TEST_CASE	(	TensorInfoAssignmentOperator	,
		TensorInfoFixture
	)

Definition at line 93 of file TensorTest.cpp.

{
    TensorInfo copy;
    copy = m_TensorInfo;
    BOOST_TEST(copy == m_TensorInfo);
}

CheckTensor()

void CheckTensor

(

const ConstTensor &

t

)

Definition at line 100 of file TensorTest.cpp.

References BaseTensor< MemoryType >::GetInfo().

Referenced by BOOST_AUTO_TEST_CASE().

{
    t.GetInfo();
}