diff options
author | Cathal Corbett <cathal.corbett@arm.com> | 2022-12-20 18:35:34 +0000 |
---|---|---|
committer | Cathal Corbett <cathal.corbett@arm.com> | 2022-12-21 15:12:02 +0000 |
commit | 0637bf38b24bba3a3d88f34ed956111a3abddda2 (patch) | |
tree | 81b348da58714cb817fcf6ab3bb31368b56ca8b7 /test/1.1/Mean.cpp | |
parent | 8de96f7e72e212503dadb450ccdcd15b411a7a1a (diff) | |
download | android-nn-driver-0637bf38b24bba3a3d88f34ed956111a3abddda2.tar.gz |
IVGCVSW-7211 Fix float16 operators being wrongly unsupported with android-nn-driver.
!armnn:8862
* Added float16 mean test cases.
* Float16 CTS/CTS pass on CpuAcc.
Signed-off-by: Cathal Corbett <cathal.corbett@arm.com>
Change-Id: Ibd9021d0ae4a205cc2c91555f3ae00c6dba84609
Diffstat (limited to 'test/1.1/Mean.cpp')
-rw-r--r-- | test/1.1/Mean.cpp | 14 |
1 files changed, 7 insertions, 7 deletions
diff --git a/test/1.1/Mean.cpp b/test/1.1/Mean.cpp index 34c29bad..70bdc3d3 100644 --- a/test/1.1/Mean.cpp +++ b/test/1.1/Mean.cpp @@ -1,5 +1,5 @@ // -// Copyright © 2017 Arm Ltd and Contributors. All rights reserved. +// Copyright © 2017, 2022 Arm Ltd and Contributors. All rights reserved. // SPDX-License-Identifier: MIT // @@ -124,7 +124,7 @@ DOCTEST_TEST_SUITE("MeanTests_CpuRef") MeanTestImpl(input, axisDimensions, axisValues, keepDims, expectedOutput, false, armnn::Compute::CpuRef); } - DOCTEST_TEST_CASE("MeanFp16NoKeepDimsTest_CpuRef") + DOCTEST_TEST_CASE("MeanFp16EnabledNoKeepDimsTest_CpuRef") { TestTensor input{ armnn::TensorShape{ 4, 3, 2 }, { 1.0f, 2.0f, 3.0f, 4.0f, 5.0f, 6.0f, 7.0f, 8.0f, 9.0f, 10.0f, @@ -138,7 +138,7 @@ DOCTEST_TEST_SUITE("MeanTests_CpuRef") MeanTestImpl(input, axisDimensions, axisValues, keepDims, expectedOutput, true, armnn::Compute::CpuRef); } - DOCTEST_TEST_CASE("MeanFp16KeepDimsTest_CpuRef") + DOCTEST_TEST_CASE("MeanFp16EnabledKeepDimsTest_CpuRef") { TestTensor input{ armnn::TensorShape{ 1, 1, 3, 2 }, { 1.0f, 1.0f, 2.0f, 2.0f, 3.0f, 3.0f } }; hidl_vec<uint32_t> axisDimensions = { 1 }; @@ -179,7 +179,7 @@ DOCTEST_TEST_SUITE("MeanTests_CpuAcc") MeanTestImpl(input, axisDimensions, axisValues, keepDims, expectedOutput, false, armnn::Compute::CpuAcc); } - DOCTEST_TEST_CASE("MeanFp16NoKeepDimsTest_CpuAcc") + DOCTEST_TEST_CASE("MeanFp16EnabledNoKeepDimsTest_CpuAcc") { TestTensor input{ armnn::TensorShape{ 4, 3, 2 }, { 1.0f, 2.0f, 3.0f, 4.0f, 5.0f, 6.0f, 7.0f, 8.0f, 9.0f, 10.0f, @@ -193,7 +193,7 @@ DOCTEST_TEST_SUITE("MeanTests_CpuAcc") MeanTestImpl(input, axisDimensions, axisValues, keepDims, expectedOutput, true, armnn::Compute::CpuAcc); } - DOCTEST_TEST_CASE("MeanFp16KeepDimsTest_CpuAcc") + DOCTEST_TEST_CASE("MeanFp16EnabledKeepDimsTest_CpuAcc") { TestTensor input{ armnn::TensorShape{ 1, 1, 3, 2 }, { 1.0f, 1.0f, 2.0f, 2.0f, 3.0f, 3.0f } }; hidl_vec<uint32_t> axisDimensions = { 1 }; @@ -232,7 +232,7 @@ DOCTEST_TEST_SUITE("MeanTests_GpuAcc") MeanTestImpl(input, axisDimensions, axisValues, keepDims, expectedOutput, false, armnn::Compute::GpuAcc); } - DOCTEST_TEST_CASE("MeanFp16NoKeepDimsTest_GpuAcc") + DOCTEST_TEST_CASE("MeanFp16EnabledNoKeepDimsTest_GpuAcc") { TestTensor input{ armnn::TensorShape{ 4, 3, 2 }, { 1.0f, 2.0f, 3.0f, 4.0f, 5.0f, 6.0f, 7.0f, 8.0f, 9.0f, 10.0f, @@ -246,7 +246,7 @@ DOCTEST_TEST_SUITE("MeanTests_GpuAcc") MeanTestImpl(input, axisDimensions, axisValues, keepDims, expectedOutput, true, armnn::Compute::GpuAcc); } - DOCTEST_TEST_CASE("MeanFp16KeepDimsTest_GpuAcc") + DOCTEST_TEST_CASE("MeanFp16EnabledKeepDimsTest_GpuAcc") { TestTensor input{ armnn::TensorShape{ 1, 1, 3, 2 }, { 1.0f, 1.0f, 2.0f, 2.0f, 3.0f, 3.0f } }; hidl_vec<uint32_t> axisDimensions = { 1 }; |