diff options
Diffstat (limited to 'delegate/test/TransposeConvolution2dTest.cpp')
| -rw-r--r-- | delegate/test/TransposeConvolution2dTest.cpp | 58 |
1 files changed, 10 insertions, 48 deletions
diff --git a/delegate/test/TransposeConvolution2dTest.cpp b/delegate/test/TransposeConvolution2dTest.cpp index b5261921f9..7c3728cedf 100644 --- a/delegate/test/TransposeConvolution2dTest.cpp +++ b/delegate/test/TransposeConvolution2dTest.cpp @@ -18,7 +18,7 @@ namespace armnnDelegate { -void TransposeConvInt8Test(std::vector<armnn::BackendId>& backends) +void TransposeConvInt8Test() { // Set input data std::vector<int32_t> transposeTensorShape { 4 }; @@ -37,8 +37,7 @@ void TransposeConvInt8Test(std::vector<armnn::BackendId>& backends) }; tflite::Padding padding = tflite::Padding_VALID; - TransposeConvTest<int8_t>(backends, - ::tflite::TensorType_INT8, + TransposeConvTest<int8_t>(::tflite::TensorType_INT8, 1, // strideX 1, // strideY padding, @@ -52,7 +51,7 @@ void TransposeConvInt8Test(std::vector<armnn::BackendId>& backends) expectedOutputValues); } -void TransposeConvFp32Test(std::vector<armnn::BackendId>& backends) +void TransposeConvFp32Test() { std::vector<int32_t> transposeTensorShape { 4 }; std::vector<int32_t> filterShape { 1, 2, 2, 1 }; @@ -70,8 +69,7 @@ void TransposeConvFp32Test(std::vector<armnn::BackendId>& backends) }; tflite::Padding padding = tflite::Padding_VALID; - TransposeConvTest<float>(backends, - ::tflite::TensorType_FLOAT32, + TransposeConvTest<float>(::tflite::TensorType_FLOAT32, 1, // strideX 1, // strideY padding, @@ -85,55 +83,19 @@ void TransposeConvFp32Test(std::vector<armnn::BackendId>& backends) expectedOutputValues); } -TEST_SUITE("TransposeConv_CpuRef_Test") +TEST_SUITE("TransposeConv_Test") { -TEST_CASE ("TransposeConv_CpuRef_Fp32_Test") +TEST_CASE ("TransposeConv_Fp32_Test") { - std::vector <armnn::BackendId> backends = {armnn::Compute::CpuRef}; - TransposeConvFp32Test(backends); + TransposeConvFp32Test(); } -TEST_CASE ("TransposeConv_CpuRef_Int8_Test") +TEST_CASE ("TransposeConv_Int8_Test") { - std::vector <armnn::BackendId> backends = {armnn::Compute::CpuRef}; - TransposeConvInt8Test(backends); + TransposeConvInt8Test(); } -} // End of TEST_SUITE(TransposeConv_CpuRef_Test) - -TEST_SUITE("TransposeConv_CpuAcc_Test") -{ - -TEST_CASE ("TransposeConv_CpuAcc_Fp32_Test") -{ - std::vector <armnn::BackendId> backends = {armnn::Compute::CpuAcc}; - TransposeConvFp32Test(backends); -} - -TEST_CASE ("TransposeConv_CpuAcc_Int8_Test") -{ - std::vector <armnn::BackendId> backends = {armnn::Compute::CpuAcc}; - TransposeConvInt8Test(backends); -} - -} // End of TEST_SUITE(TransposeConv_CpuAcc_Test) - -TEST_SUITE("TransposeConv_GpuAcc_Test") -{ - -TEST_CASE ("TransposeConv_GpuAcc_Fp32_Test") -{ - std::vector <armnn::BackendId> backends = {armnn::Compute::GpuAcc}; - TransposeConvFp32Test(backends); -} - -TEST_CASE ("TransposeConv_GpuAcc_Int8_Test") -{ - std::vector <armnn::BackendId> backends = {armnn::Compute::GpuAcc}; - TransposeConvInt8Test(backends); -} - -} // End of TEST_SUITE(TransposeConv_GpuAcc_Test) +} // End of TEST_SUITE(TransposeConv_Test) } // namespace armnnDelegate
\ No newline at end of file |