blob: 4ab748806c713b0510a33c07b64ce243bd93aa98 [file] [log] [blame]
Aron Virginas-Tar13b653f2019-11-01 11:40:39 +00001//
2// Copyright © 2019 Arm Ltd. All rights reserved.
3// SPDX-License-Identifier: MIT
4//
5
6#include <aclCommon/ArmComputeTensorUtils.hpp>
7
8#include <boost/test/unit_test.hpp>
9
10using namespace armnn::armcomputetensorutils;
11
12BOOST_AUTO_TEST_SUITE(ArmComputeTensorUtils)
13
14BOOST_AUTO_TEST_CASE(BuildArmComputeTensorInfoTest)
15{
16
17 const armnn::TensorShape tensorShape = { 1, 2, 3, 4 };
Derek Lambertif90c56d2020-01-10 17:14:08 +000018 const armnn::DataType dataType = armnn::DataType::QAsymmU8;
Aron Virginas-Tar13b653f2019-11-01 11:40:39 +000019
20 const std::vector<float> quantScales = { 1.5f, 2.5f, 3.5f, 4.5f };
21 const float quantScale = quantScales[0];
22 const int32_t quantOffset = 128;
23
24 // Tensor info with per-tensor quantization
25 const armnn::TensorInfo tensorInfo0(tensorShape, dataType, quantScale, quantOffset);
26 const arm_compute::TensorInfo aclTensorInfo0 = BuildArmComputeTensorInfo(tensorInfo0);
27
28 const arm_compute::TensorShape& aclTensorShape = aclTensorInfo0.tensor_shape();
29 BOOST_CHECK(aclTensorShape.num_dimensions() == tensorShape.GetNumDimensions());
30 for(unsigned int i = 0u; i < tensorShape.GetNumDimensions(); ++i)
31 {
32 // NOTE: arm_compute tensor dimensions are stored in the opposite order
33 BOOST_CHECK(aclTensorShape[i] == tensorShape[tensorShape.GetNumDimensions() - i - 1]);
34 }
35
36 BOOST_CHECK(aclTensorInfo0.data_type() == arm_compute::DataType::QASYMM8);
37 BOOST_CHECK(aclTensorInfo0.quantization_info().scale()[0] == quantScale);
38
39 // Tensor info with per-axis quantization
40 const armnn::TensorInfo tensorInfo1(tensorShape, dataType, quantScales, 0);
41 const arm_compute::TensorInfo aclTensorInfo1 = BuildArmComputeTensorInfo(tensorInfo1);
42
43 BOOST_CHECK(aclTensorInfo1.quantization_info().scale() == quantScales);
44}
45
46BOOST_AUTO_TEST_SUITE_END()