diff options
Diffstat (limited to 'src/backends/backendsCommon/test/layerTests/BatchMatMulTestImpl.cpp')
-rw-r--r-- | src/backends/backendsCommon/test/layerTests/BatchMatMulTestImpl.cpp | 169 |
1 files changed, 13 insertions, 156 deletions
diff --git a/src/backends/backendsCommon/test/layerTests/BatchMatMulTestImpl.cpp b/src/backends/backendsCommon/test/layerTests/BatchMatMulTestImpl.cpp index 6fcc35ab52..74bd97f103 100644 --- a/src/backends/backendsCommon/test/layerTests/BatchMatMulTestImpl.cpp +++ b/src/backends/backendsCommon/test/layerTests/BatchMatMulTestImpl.cpp @@ -71,20 +71,9 @@ LayerTestResult<T, 2> BatchMatMul2DSimpleTest( { auto descriptor = armnn::BatchMatMulDescriptor(); // Arbitrary layout with no transpose/adjointing - float qScale = 0.0f; + float qScale = 1.0f; int32_t qOffset = 0; - switch(ArmnnType) - { - case armnn::DataType::QAsymmS8: - case armnn::DataType::QAsymmU8: - case armnn::DataType::QSymmS16: - qScale = 1.0f; - break; - default: - break; - } - armnn::TensorInfo inputXInfo({2,2}, ArmnnType, qScale, qOffset); armnn::TensorInfo inputYInfo({2,2}, ArmnnType, qScale, qOffset); armnn::TensorInfo outputInfo({2,2}, ArmnnType, qScale, qOffset); @@ -160,20 +149,9 @@ LayerTestResult<T, 3> BatchMatMul3DSimpleTest( { auto descriptor = armnn::BatchMatMulDescriptor(); // Arbitrary layout with no transpose/adjointing - float qScale = 0.0f; + float qScale = 1.0f; int32_t qOffset = 0; - switch(ArmnnType) - { - case armnn::DataType::QAsymmS8: - case armnn::DataType::QAsymmU8: - case armnn::DataType::QSymmS16: - qScale = 1.0f; - break; - default: - break; - } - armnn::TensorInfo inputXInfo({1,2,2}, ArmnnType, qScale, qOffset); armnn::TensorInfo inputYInfo({1,2,2}, ArmnnType, qScale, qOffset); armnn::TensorInfo outputInfo({1,2,2}, ArmnnType, qScale, qOffset); @@ -249,20 +227,9 @@ LayerTestResult<T, 4> BatchMatMulNCHWSimpleTest( { auto descriptor = armnn::BatchMatMulDescriptor(); // Default arbitrary layout is treated the same as NCHW - float qScale = 0.0f; + float qScale = 1.0f; int32_t qOffset = 0; - switch(ArmnnType) - { - case armnn::DataType::QAsymmS8: - case armnn::DataType::QAsymmU8: - case armnn::DataType::QSymmS16: - qScale = 1.0f; - break; - default: - break; - } - armnn::TensorInfo inputXInfo({1,1,2,2}, ArmnnType, qScale, qOffset); armnn::TensorInfo inputYInfo({1,1,2,2}, ArmnnType, qScale, qOffset); armnn::TensorInfo outputInfo({1,1,2,2}, ArmnnType, qScale, qOffset); @@ -343,20 +310,9 @@ LayerTestResult<T, 4> BatchMatMulNHWCSimpleTest( armnn::DataLayout::NHWC, armnn::DataLayout::NHWC); - float qScale = 0.0f; + float qScale = 1.0f; int32_t qOffset = 0; - switch(ArmnnType) - { - case armnn::DataType::QAsymmS8: - case armnn::DataType::QAsymmU8: - case armnn::DataType::QSymmS16: - qScale = 1.0f; - break; - default: - break; - } - armnn::TensorInfo inputXInfo({1,2,2,1}, ArmnnType, qScale, qOffset); armnn::TensorInfo inputYInfo({1,2,2,1}, ArmnnType, qScale, qOffset); armnn::TensorInfo outputInfo({1,2,2,1}, ArmnnType, qScale, qOffset); @@ -432,20 +388,9 @@ LayerTestResult<T, 3> BatchMatMul3DBatchTest( { auto descriptor = armnn::BatchMatMulDescriptor(); // Arbitrary layout with no transpose/adjointing - float qScale = 0.0f; + float qScale = 1.0f; int32_t qOffset = 0; - switch(ArmnnType) - { - case armnn::DataType::QAsymmS8: - case armnn::DataType::QAsymmU8: - case armnn::DataType::QSymmS16: - qScale = 1.0f; - break; - default: - break; - } - armnn::TensorInfo inputXInfo({2,2,2}, ArmnnType, qScale, qOffset); armnn::TensorInfo inputYInfo({2,2,2}, ArmnnType, qScale, qOffset); armnn::TensorInfo outputInfo({2,2,2}, ArmnnType, qScale, qOffset); @@ -530,20 +475,9 @@ LayerTestResult<T, 3> BatchMatMul3DBroadcastTest( { auto descriptor = armnn::BatchMatMulDescriptor(); // Arbitrary layout with no transpose/adjointing - float qScale = 0.0f; + float qScale = 1.0f; int32_t qOffset = 0; - switch(ArmnnType) - { - case armnn::DataType::QAsymmS8: - case armnn::DataType::QAsymmU8: - case armnn::DataType::QSymmS16: - qScale = 1.0f; - break; - default: - break; - } - armnn::TensorInfo inputXInfo({2,2,2}, ArmnnType, qScale, qOffset); armnn::TensorInfo inputYInfo({1,2,2}, ArmnnType, qScale, qOffset); armnn::TensorInfo outputInfo({2,2,2}, ArmnnType, qScale, qOffset); @@ -625,20 +559,9 @@ LayerTestResult<T, 3> BatchMatMul3D2DBroadcastTest( { auto descriptor = armnn::BatchMatMulDescriptor(); // Arbitrary layout with no transpose/adjointing - float qScale = 0.0f; + float qScale = 1.0f; int32_t qOffset = 0; - switch(ArmnnType) - { - case armnn::DataType::QAsymmS8: - case armnn::DataType::QAsymmU8: - case armnn::DataType::QSymmS16: - qScale = 1.0f; - break; - default: - break; - } - armnn::TensorInfo inputXInfo({2,2,2}, ArmnnType, qScale, qOffset); armnn::TensorInfo inputYInfo({2,2}, ArmnnType, qScale, qOffset); armnn::TensorInfo outputInfo({2,2,2}, ArmnnType, qScale, qOffset); @@ -725,20 +648,9 @@ LayerTestResult<T, 5> BatchMatMulNDHWCNHWCTest( armnn::DataLayout::NDHWC, armnn::DataLayout::NHWC); - float qScale = 0.0f; + float qScale = 1.0f; int32_t qOffset = 0; - switch(ArmnnType) - { - case armnn::DataType::QAsymmS8: - case armnn::DataType::QAsymmU8: - case armnn::DataType::QSymmS16: - qScale = 1.0f; - break; - default: - break; - } - armnn::TensorInfo inputXInfo({1,1,2,2,2}, ArmnnType, qScale, qOffset); armnn::TensorInfo inputYInfo({1,2,2,2}, ArmnnType, qScale, qOffset); armnn::TensorInfo outputInfo({1,1,2,2,2}, ArmnnType, qScale, qOffset); @@ -823,20 +735,9 @@ LayerTestResult<T, 2> BatchMatMul2DTinyTest( { auto descriptor = armnn::BatchMatMulDescriptor(); // Arbitrary layout with no transpose/adjointing - float qScale = 0.0f; + float qScale = 1.0f; int32_t qOffset = 0; - switch(ArmnnType) - { - case armnn::DataType::QAsymmS8: - case armnn::DataType::QAsymmU8: - case armnn::DataType::QSymmS16: - qScale = 1.0f; - break; - default: - break; - } - armnn::TensorInfo inputXInfo({1,1}, ArmnnType, qScale, qOffset); armnn::TensorInfo inputYInfo({1,1}, ArmnnType, qScale, qOffset); armnn::TensorInfo outputInfo({1,1}, ArmnnType, qScale, qOffset); @@ -909,20 +810,9 @@ LayerTestResult<T, 3> BatchMatMul3DNonSquareTest( { auto descriptor = armnn::BatchMatMulDescriptor(); // Arbitrary layout with no transpose/adjointing - float qScale = 0.0f; + float qScale = 1.0f; int32_t qOffset = 0; - switch(ArmnnType) - { - case armnn::DataType::QAsymmS8: - case armnn::DataType::QAsymmU8: - case armnn::DataType::QSymmS16: - qScale = 1.0f; - break; - default: - break; - } - armnn::TensorInfo inputXInfo({2,5,3}, ArmnnType, qScale, qOffset); armnn::TensorInfo inputYInfo({2,3,4}, ArmnnType, qScale, qOffset); armnn::TensorInfo outputInfo({2,5,4}, ArmnnType, qScale, qOffset); @@ -1024,20 +914,9 @@ LayerTestResult<T, 2> BatchMatMul2DTranspSimpleTest( false, false); - float qScale = 0.0f; + float qScale = 1.0f; int32_t qOffset = 0; - switch(ArmnnType) - { - case armnn::DataType::QAsymmS8: - case armnn::DataType::QAsymmU8: - case armnn::DataType::QSymmS16: - qScale = 1.0f; - break; - default: - break; - } - armnn::TensorInfo inputXInfo({2,3}, ArmnnType, qScale, qOffset); armnn::TensorInfo inputYInfo({2,3}, ArmnnType, qScale, qOffset); armnn::TensorInfo outputInfo({3,3}, ArmnnType, qScale, qOffset); @@ -1117,20 +996,9 @@ LayerTestResult<T, 2> BatchMatMul2DAdjointSimpleTest( true, false); - float qScale = 0.0f; + float qScale = 1.0f; int32_t qOffset = 0; - switch(ArmnnType) - { - case armnn::DataType::QAsymmS8: - case armnn::DataType::QAsymmU8: - case armnn::DataType::QSymmS16: - qScale = 1.0f; - break; - default: - break; - } - armnn::TensorInfo inputXInfo({3,3}, ArmnnType, qScale, qOffset); armnn::TensorInfo inputYInfo({3,3}, ArmnnType, qScale, qOffset); armnn::TensorInfo outputInfo({3,3}, ArmnnType, qScale, qOffset); @@ -1227,20 +1095,9 @@ LayerTestResult<T, 4> BatchMatMulNHWCParamsTest( armnn::DataLayout::NHWC, armnn::DataLayout::NHWC); - float qScale = 0.0f; + float qScale = 1.0f; int32_t qOffset = 0; - switch(ArmnnType) - { - case armnn::DataType::QAsymmS8: - case armnn::DataType::QAsymmU8: - case armnn::DataType::QSymmS16: - qScale = 1.0f; - break; - default: - break; - } - armnn::TensorInfo inputXInfo({1,4,4,2}, ArmnnType, qScale, qOffset); armnn::TensorInfo inputYInfo({2,2,4,1}, ArmnnType, qScale, qOffset); armnn::TensorInfo outputInfo({2,4,2,2}, ArmnnType, qScale, qOffset); |