From 7afc92c7808d0ac804d4baff8921001c86efcd21 Mon Sep 17 00:00:00 2001 From: Matthew Sloyan Date: Thu, 4 Nov 2021 16:52:34 +0000 Subject: Add CpuAcc and GpuAcc TfLite Delegate tests for Conv3d Signed-off-by: Matthew Sloyan Change-Id: I3e91796a69f02a8eff3018a1d17a496a66076db5 --- delegate/src/test/Convolution3dTest.cpp | 47 ++++++++++++++++++++++++++++++++- 1 file changed, 46 insertions(+), 1 deletion(-) 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 backends = {armnn::Compute::CpuAcc}; + Conv3DWithBiasesSimpleWithPaddingFp32Test(backends); +} + +TEST_CASE ("Conv3DWithBiasesStrides_Fp32_CpuAcc_Test") +{ + std::vector backends = {armnn::Compute::CpuAcc}; + Conv3DWithBiasesStridesFp32Test(backends); +} + +TEST_CASE ("Conv3DFp32Small_Fp32_CpuAcc_Test") +{ + std::vector backends = {armnn::Compute::CpuAcc}; + Conv3DFp32SmallTest(backends); +} + +} //End of TEST_SUITE("Convolution3dTest_CpuAccTests") + +TEST_SUITE("Convolution3dTest_GpuAccTests") +{ + +TEST_CASE ("Conv3DWithBiasesSimpleWithPadding_Fp32_GpuAcc_Test") +{ + std::vector backends = {armnn::Compute::GpuAcc}; + Conv3DWithBiasesSimpleWithPaddingFp32Test(backends); +} + +TEST_CASE ("Conv3DWithBiasesStrides_Fp32_GpuAcc_Test") +{ + std::vector backends = {armnn::Compute::GpuAcc}; + Conv3DWithBiasesStridesFp32Test(backends); +} + +TEST_CASE ("Conv3DFp32Small_Fp32_GpuAcc_Test") +{ + std::vector backends = {armnn::Compute::GpuAcc}; + Conv3DFp32SmallTest(backends); +} + +} //End of TEST_SUITE("Convolution3dTest_GpuAccTests") + #endif } // namespace armnnDelegate \ No newline at end of file -- cgit v1.2.1