diff options
author | John Mcloughlin <john.mcloughlin@arm.com> | 2024-01-31 11:00:27 +0000 |
---|---|---|
committer | John Mcloughlin <john.mcloughlin@arm.com> | 2024-01-31 21:24:23 +0000 |
commit | 829e13edbe9487b5a0600688cec6312b867e2f07 (patch) | |
tree | 2b9176173e73afc1f1cdf62cec334c7ee64e7123 /src/backends/gpuFsa/test/GpuFsaEndToEndTests.cpp | |
parent | e4134fc28ce2a9c751fb22ff84b733d067255849 (diff) | |
download | armnn-829e13edbe9487b5a0600688cec6312b867e2f07.tar.gz |
IVGCVSW-7568 Implement Sub ElementwiseBinary operator GpuFsa
* Added support for Gpu Sub operator
* Added unit tests
Signed-off-by: John Mcloughlin <john.mcloughlin@arm.com>
Change-Id: I1efaa485772a3716e3781566843bd50bd9bab811
Diffstat (limited to 'src/backends/gpuFsa/test/GpuFsaEndToEndTests.cpp')
-rw-r--r-- | src/backends/gpuFsa/test/GpuFsaEndToEndTests.cpp | 11 |
1 files changed, 11 insertions, 0 deletions
diff --git a/src/backends/gpuFsa/test/GpuFsaEndToEndTests.cpp b/src/backends/gpuFsa/test/GpuFsaEndToEndTests.cpp index 9972b24637..26c7cb8d63 100644 --- a/src/backends/gpuFsa/test/GpuFsaEndToEndTests.cpp +++ b/src/backends/gpuFsa/test/GpuFsaEndToEndTests.cpp @@ -45,4 +45,15 @@ TEST_CASE("GpuFsaElementwiseBinaryAddTestFloat16") ElementwiseBinarySimple3DEndToEnd<armnn::DataType::Float16>(gpuFsaDefaultBackends, BinaryOperation::Add); } +// ElementwiseBinary Sub +TEST_CASE("GpuFsaElementwiseBinarySubTestFloat32") +{ + ElementwiseBinarySimple3DEndToEnd<armnn::DataType::Float32>(gpuFsaDefaultBackends, BinaryOperation::Sub); +} + +TEST_CASE("GpuFsaElementwiseBinarySubTestFloat16") +{ + ElementwiseBinarySimple3DEndToEnd<armnn::DataType::Float16>(gpuFsaDefaultBackends, BinaryOperation::Sub); +} + } |