diff options
author | Matthew Sloyan <matthew.sloyan@arm.com> | 2021-11-04 16:52:34 +0000 |
---|---|---|
committer | Matthew Sloyan <matthew.sloyan@arm.com> | 2021-11-04 16:57:18 +0000 |
commit | 7afc92c7808d0ac804d4baff8921001c86efcd21 (patch) | |
tree | 87dd4f1c9f367481852d0e4247cf05454c75625d /delegate/src | |
parent | f9d96e5a04810b4f1182b4c1b0f27601f6deb0dd (diff) | |
download | armnn-7afc92c7808d0ac804d4baff8921001c86efcd21.tar.gz |
Add CpuAcc and GpuAcc TfLite Delegate tests for Conv3d
Signed-off-by: Matthew Sloyan <matthew.sloyan@arm.com>
Change-Id: I3e91796a69f02a8eff3018a1d17a496a66076db5
Diffstat (limited to 'delegate/src')
-rw-r--r-- | delegate/src/test/Convolution3dTest.cpp | 47 |
1 files changed, 46 insertions, 1 deletions
diff --git a/delegate/src/test/Convolution3dTest.cpp b/delegate/src/test/Convolution3dTest.cpp index 6caa7ea18f..06883f186d 100644 --- a/delegate/src/test/Convolution3dTest.cpp +++ b/delegate/src/test/Convolution3dTest.cpp @@ -265,9 +265,54 @@ TEST_CASE ("Conv3DFp32Small_Fp32_CpuRef_Test") Conv3DFp32SmallTest(backends); } - } //End of TEST_SUITE("Convolution3dTest_CpuRefTests") +TEST_SUITE("Convolution3dTest_CpuAccTests") +{ + +TEST_CASE ("Conv3DWithBiasesSimpleWithPadding_Fp32_CpuAcc_Test") +{ + std::vector <armnn::BackendId> backends = {armnn::Compute::CpuAcc}; + Conv3DWithBiasesSimpleWithPaddingFp32Test(backends); +} + +TEST_CASE ("Conv3DWithBiasesStrides_Fp32_CpuAcc_Test") +{ + std::vector <armnn::BackendId> backends = {armnn::Compute::CpuAcc}; + Conv3DWithBiasesStridesFp32Test(backends); +} + +TEST_CASE ("Conv3DFp32Small_Fp32_CpuAcc_Test") +{ + std::vector <armnn::BackendId> backends = {armnn::Compute::CpuAcc}; + Conv3DFp32SmallTest(backends); +} + +} //End of TEST_SUITE("Convolution3dTest_CpuAccTests") + +TEST_SUITE("Convolution3dTest_GpuAccTests") +{ + +TEST_CASE ("Conv3DWithBiasesSimpleWithPadding_Fp32_GpuAcc_Test") +{ + std::vector <armnn::BackendId> backends = {armnn::Compute::GpuAcc}; + Conv3DWithBiasesSimpleWithPaddingFp32Test(backends); +} + +TEST_CASE ("Conv3DWithBiasesStrides_Fp32_GpuAcc_Test") +{ + std::vector <armnn::BackendId> backends = {armnn::Compute::GpuAcc}; + Conv3DWithBiasesStridesFp32Test(backends); +} + +TEST_CASE ("Conv3DFp32Small_Fp32_GpuAcc_Test") +{ + std::vector <armnn::BackendId> backends = {armnn::Compute::GpuAcc}; + Conv3DFp32SmallTest(backends); +} + +} //End of TEST_SUITE("Convolution3dTest_GpuAccTests") + #endif } // namespace armnnDelegate
\ No newline at end of file |