8 #include <doctest/doctest.h> 14 TEST_CASE(
"BuildArmComputeTensorInfoTest")
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;
25 const armnn::TensorInfo tensorInfo0(tensorShape, dataType, quantScale, quantOffset);
26 const arm_compute::TensorInfo aclTensorInfo0 = BuildArmComputeTensorInfo(tensorInfo0);
28 const arm_compute::TensorShape& aclTensorShape = aclTensorInfo0.tensor_shape();
33 CHECK(aclTensorShape[i] == tensorShape[tensorShape.
GetNumDimensions() - i - 1]);
36 CHECK(aclTensorInfo0.data_type() == arm_compute::DataType::QASYMM8);
37 CHECK(aclTensorInfo0.quantization_info().scale()[0] == quantScale);
41 const arm_compute::TensorInfo aclTensorInfo1 = BuildArmComputeTensorInfo(tensorInfo1);
43 CHECK(aclTensorInfo1.quantization_info().scale() == quantScales);
TEST_SUITE("TestConstTensorLayerVisitor")
unsigned int GetNumDimensions() const
Function that returns the tensor rank.