diff options
Diffstat (limited to 'delegate/test/ElementwiseUnaryTest.cpp')
-rw-r--r-- | delegate/test/ElementwiseUnaryTest.cpp | 304 |
1 files changed, 20 insertions, 284 deletions
diff --git a/delegate/test/ElementwiseUnaryTest.cpp b/delegate/test/ElementwiseUnaryTest.cpp index 14a6061191..13d703913a 100644 --- a/delegate/test/ElementwiseUnaryTest.cpp +++ b/delegate/test/ElementwiseUnaryTest.cpp @@ -18,13 +18,11 @@ namespace armnnDelegate { -TEST_SUITE("ElementwiseUnary_GpuAccTests") +TEST_SUITE("ElementwiseUnary_Tests") { -TEST_CASE ("Abs_Float32_GpuAcc_Test") +TEST_CASE ("Abs_Float32_Test") { - // Create the ArmNN Delegate - std::vector<armnn::BackendId> backends = { armnn::Compute::GpuAcc }; // Set input data std::vector<float> inputValues { @@ -37,266 +35,13 @@ TEST_CASE ("Abs_Float32_GpuAcc_Test") { expectedOutputValues[i] = std::abs(inputValues[i]); } - ElementwiseUnaryFP32Test(tflite::BuiltinOperator_ABS, backends, inputValues, expectedOutputValues); -} - -TEST_CASE ("Exp_Float32_GpuAcc_Test") -{ - // Create the ArmNN Delegate - std::vector<armnn::BackendId> backends = { armnn::Compute::GpuAcc }; - // Set input data - std::vector<float> inputValues - { - 5.0f, 4.0f, - 3.0f, 2.0f, - 1.0f, 1.1f - }; - // Set output data - std::vector<float> expectedOutputValues - { - 148.413159102577f, 54.598150033144f, - 20.085536923188f, 7.389056098931f, - 2.718281828459f, 3.004166023946f - }; - - ElementwiseUnaryFP32Test(tflite::BuiltinOperator_EXP, backends, inputValues, expectedOutputValues); -} - -TEST_CASE ("Log_Float32_GpuAcc_Test") -{ - // Create the ArmNN Delegate - std::vector<armnn::BackendId> backends = { armnn::Compute::GpuAcc }; - // Set input data - std::vector<float> inputValues - { - 1.0f, 1.0f, 2.0f, - 3.0f, 4.0f, 2.71828f - }; - // Set output data - std::vector<float> expectedOutputValues - { - 0.f, 0.f, 0.69314718056f, - 1.09861228867f, 1.38629436112f, 0.99999932734f - }; - - ElementwiseUnaryFP32Test(tflite::BuiltinOperator_LOG, backends, inputValues, expectedOutputValues); -} - -TEST_CASE ("Neg_Float32_GpuAcc_Test") -{ - // Create the ArmNN Delegate - std::vector<armnn::BackendId> backends = { armnn::Compute::GpuAcc }; - // Set input data - std::vector<float> inputValues - { - 1.f, 0.f, 3.f, - 25.f, 64.f, 100.f - }; - // Set output data - std::vector<float> expectedOutputValues - { - -1.f, 0.f, -3.f, - -25.f, -64.f, -100.f - }; - - ElementwiseUnaryFP32Test(tflite::BuiltinOperator_NEG, backends, inputValues, expectedOutputValues); -} -TEST_CASE ("Rsqrt_Float32_GpuAcc_Test") -{ - // Create the ArmNN Delegate - std::vector<armnn::BackendId> backends = { armnn::Compute::GpuAcc }; - // Set input data - std::vector<float> inputValues - { - 1.f, 4.f, 16.f, - 25.f, 64.f, 100.f - }; - // Set output data - std::vector<float> expectedOutputValues - { - 1.f, 0.5f, 0.25f, - 0.2f, 0.125f, 0.1f - }; - - ElementwiseUnaryFP32Test(tflite::BuiltinOperator_RSQRT, backends, inputValues, expectedOutputValues); + ElementwiseUnaryFP32Test(tflite::BuiltinOperator_ABS, inputValues, expectedOutputValues); } -TEST_CASE ("Sin_Float32_GpuAcc_Test") +TEST_CASE ("Ceil_Float32_Test") { - // Create the ArmNN Delegate - std::vector<armnn::BackendId> backends = { armnn::Compute::GpuAcc }; - // Set input data - std::vector<float> inputValues - { - 0.0f, 1.0f, 16.0f, - 0.5f, 36.0f, -1.f - }; - // Set output data - std::vector<float> expectedOutputValues - { - 0.0f, 0.8414709848f, -0.28790331666f, - 0.4794255386f, -0.99177885344f, -0.8414709848f - }; - - ElementwiseUnaryFP32Test(tflite::BuiltinOperator_SIN, backends, inputValues, expectedOutputValues); -} -} // TEST_SUITE("ElementwiseUnary_GpuAccTests") - - - -TEST_SUITE("ElementwiseUnary_CpuAccTests") -{ - -TEST_CASE ("Abs_Float32_CpuAcc_Test") -{ - // Create the ArmNN Delegate - std::vector<armnn::BackendId> backends = { armnn::Compute::CpuAcc }; - // Set input data - std::vector<float> inputValues - { - -0.1f, -0.2f, -0.3f, - 0.1f, 0.2f, 0.3f - }; - // Calculate output data - std::vector<float> expectedOutputValues(inputValues.size()); - for (unsigned int i = 0; i < inputValues.size(); ++i) - { - expectedOutputValues[i] = std::abs(inputValues[i]); - } - - ElementwiseUnaryFP32Test(tflite::BuiltinOperator_ABS, backends, inputValues, expectedOutputValues); -} - -TEST_CASE ("Exp_Float32_CpuAcc_Test") -{ - // Create the ArmNN Delegate - std::vector<armnn::BackendId> backends = { armnn::Compute::CpuAcc }; - // Set input data - std::vector<float> inputValues - { - 5.0f, 4.0f, - 3.0f, 2.0f, - 1.0f, 1.1f - }; - // Set output data - std::vector<float> expectedOutputValues - { - 148.413159102577f, 54.598150033144f, - 20.085536923188f, 7.389056098931f, - 2.718281828459f, 3.004166023946f - }; - - ElementwiseUnaryFP32Test(tflite::BuiltinOperator_EXP, backends, inputValues, expectedOutputValues); -} - -TEST_CASE ("Log_Float32_CpuAcc_Test") -{ - // Create the ArmNN Delegate - std::vector<armnn::BackendId> backends = { armnn::Compute::CpuAcc }; - // Set input data - std::vector<float> inputValues - { - 1.0f, 1.0f, 2.0f, - 3.0f, 4.0f, 2.71828f - }; - // Set output data - std::vector<float> expectedOutputValues - { - 0.f, 0.f, 0.69314718056f, - 1.09861228867f, 1.38629436112f, 0.99999932734f - }; - - ElementwiseUnaryFP32Test(tflite::BuiltinOperator_LOG, backends, inputValues, expectedOutputValues); -} - -TEST_CASE ("Neg_Float32_CpuAcc_Test") -{ - // Create the ArmNN Delegate - std::vector<armnn::BackendId> backends = { armnn::Compute::CpuAcc }; - // Set input data - std::vector<float> inputValues - { - 1.f, 0.f, 3.f, - 25.f, 64.f, 100.f - }; - // Set output data - std::vector<float> expectedOutputValues - { - -1.f, 0.f, -3.f, - -25.f, -64.f, -100.f - }; - - ElementwiseUnaryFP32Test(tflite::BuiltinOperator_NEG, backends, inputValues, expectedOutputValues); -} - -TEST_CASE ("Rsqrt_Float32_CpuAcc_Test") -{ - // Create the ArmNN Delegate - std::vector<armnn::BackendId> backends = { armnn::Compute::CpuAcc }; - // Set input data - std::vector<float> inputValues - { - 1.f, 4.f, 16.f, - 25.f, 64.f, 100.f - }; - // Set output data - std::vector<float> expectedOutputValues - { - 1.f, 0.5f, 0.25f, - 0.2f, 0.125f, 0.1f - }; - - ElementwiseUnaryFP32Test(tflite::BuiltinOperator_RSQRT, backends, inputValues, expectedOutputValues); -} - -TEST_CASE ("Sin_Float32_CpuAcc_Test") -{ - // Create the ArmNN Delegate - std::vector<armnn::BackendId> backends = { armnn::Compute::CpuAcc }; - // Set input data - std::vector<float> inputValues - { - 0.0f, 1.0f, 16.0f, - 0.5f, 36.0f, -1.f - }; - // Set output data - std::vector<float> expectedOutputValues - { - 0.0f, 0.8414709848f, -0.28790331666f, - 0.4794255386f, -0.99177885344f, -0.8414709848f - }; - - ElementwiseUnaryFP32Test(tflite::BuiltinOperator_SIN, backends, inputValues, expectedOutputValues); -} -} // TEST_SUITE("ElementwiseUnary_CpuAccTests") - -TEST_SUITE("ElementwiseUnary_CpuRefTests") -{ - -TEST_CASE ("Abs_Float32_CpuRef_Test") -{ - // Create the ArmNN Delegate - std::vector<armnn::BackendId> backends = { armnn::Compute::CpuRef }; - // Set input data - std::vector<float> inputValues - { - -0.1f, -0.2f, -0.3f, - 0.1f, 0.2f, 0.3f - }; - // Calculate output data - std::vector<float> expectedOutputValues(inputValues.size()); - for (unsigned int i = 0; i < inputValues.size(); ++i) - { - expectedOutputValues[i] = std::abs(inputValues[i]); - } - - ElementwiseUnaryFP32Test(tflite::BuiltinOperator_ABS, backends, inputValues, expectedOutputValues); -} - -TEST_CASE ("Ceil_Float32_CpuRef_Test") -{ - // Create the ArmNN Delegate + // Only works on CpuRef std::vector<armnn::BackendId> backends = { armnn::Compute::CpuRef }; // Set input data std::vector<float> inputValues @@ -311,13 +56,11 @@ TEST_CASE ("Ceil_Float32_CpuRef_Test") 1.0f, 0.0f, -1.0f }; - ElementwiseUnaryFP32Test(tflite::BuiltinOperator_CEIL, backends, inputValues, expectedOutputValues); + ElementwiseUnaryFP32Test(tflite::BuiltinOperator_CEIL, inputValues, expectedOutputValues, backends); } -TEST_CASE ("Exp_Float32_CpuRef_Test") +TEST_CASE ("Exp_Float32_Test") { - // Create the ArmNN Delegate - std::vector<armnn::BackendId> backends = { armnn::Compute::CpuRef }; // Set input data std::vector<float> inputValues { @@ -333,13 +76,11 @@ TEST_CASE ("Exp_Float32_CpuRef_Test") 2.718281828459f, 3.004166023946f }; - ElementwiseUnaryFP32Test(tflite::BuiltinOperator_EXP, backends, inputValues, expectedOutputValues); + ElementwiseUnaryFP32Test(tflite::BuiltinOperator_EXP, inputValues, expectedOutputValues); } -TEST_CASE ("Log_Float32_CpuRef_Test") +TEST_CASE ("Log_Float32_Test") { - // Create the ArmNN Delegate - std::vector<armnn::BackendId> backends = { armnn::Compute::CpuRef }; // Set input data std::vector<float> inputValues { @@ -353,13 +94,11 @@ TEST_CASE ("Log_Float32_CpuRef_Test") 1.09861228867f, 1.38629436112f, 0.99999932734f }; - ElementwiseUnaryFP32Test(tflite::BuiltinOperator_LOG, backends, inputValues, expectedOutputValues); + ElementwiseUnaryFP32Test(tflite::BuiltinOperator_LOG, inputValues, expectedOutputValues); } -TEST_CASE ("Neg_Float32_CpuRef_Test") +TEST_CASE ("Neg_Float32_Test") { - // Create the ArmNN Delegate - std::vector<armnn::BackendId> backends = { armnn::Compute::CpuRef }; // Set input data std::vector<float> inputValues { @@ -373,13 +112,11 @@ TEST_CASE ("Neg_Float32_CpuRef_Test") -25.f, -64.f, -100.f }; - ElementwiseUnaryFP32Test(tflite::BuiltinOperator_NEG, backends, inputValues, expectedOutputValues); + ElementwiseUnaryFP32Test(tflite::BuiltinOperator_NEG, inputValues, expectedOutputValues); } -TEST_CASE ("Rsqrt_Float32_CpuRef_Test") +TEST_CASE ("Rsqrt_Float32_Test") { - // Create the ArmNN Delegate - std::vector<armnn::BackendId> backends = { armnn::Compute::CpuRef }; // Set input data std::vector<float> inputValues { @@ -393,11 +130,12 @@ TEST_CASE ("Rsqrt_Float32_CpuRef_Test") 0.2f, 0.125f, 0.1f }; - ElementwiseUnaryFP32Test(tflite::BuiltinOperator_RSQRT, backends, inputValues, expectedOutputValues); + ElementwiseUnaryFP32Test(tflite::BuiltinOperator_RSQRT, inputValues, expectedOutputValues); } -TEST_CASE ("Sqrt_Float32_CpuRef_Test") +TEST_CASE ("Sqrt_Float32_Test") { + // Only works on CpuRef. std::vector<armnn::BackendId> backends = { armnn::Compute::CpuRef }; // Set input data std::vector<float> inputValues @@ -412,13 +150,11 @@ TEST_CASE ("Sqrt_Float32_CpuRef_Test") expectedOutputValues[i] = std::sqrt(inputValues[i]); } - ElementwiseUnaryFP32Test(tflite::BuiltinOperator_SQRT, backends, inputValues, expectedOutputValues); + ElementwiseUnaryFP32Test(tflite::BuiltinOperator_SQRT, inputValues, expectedOutputValues, backends); } -TEST_CASE ("Sin_Float32_CpuRef_Test") +TEST_CASE ("Sin_Float32_Test") { - // Create the ArmNN Delegate - std::vector<armnn::BackendId> backends = { armnn::Compute::CpuRef }; // Set input data std::vector<float> inputValues { @@ -432,8 +168,8 @@ TEST_CASE ("Sin_Float32_CpuRef_Test") 0.4794255386f, -0.99177885344f, -0.8414709848f }; - ElementwiseUnaryFP32Test(tflite::BuiltinOperator_SIN, backends, inputValues, expectedOutputValues); + ElementwiseUnaryFP32Test(tflite::BuiltinOperator_SIN, inputValues, expectedOutputValues); } -} // TEST_SUITE("ElementwiseUnary_CpuRefTests") +} // TEST_SUITE("ElementwiseUnary_Tests") } // namespace armnnDelegate
\ No newline at end of file |