aboutsummaryrefslogtreecommitdiff
path: root/src/armnnTestUtils/CreateWorkload.hpp
diff options
context:
space:
mode:
Diffstat (limited to 'src/armnnTestUtils/CreateWorkload.hpp')
-rw-r--r--src/armnnTestUtils/CreateWorkload.hpp4
1 files changed, 2 insertions, 2 deletions
diff --git a/src/armnnTestUtils/CreateWorkload.hpp b/src/armnnTestUtils/CreateWorkload.hpp
index 15de5b5ddb..d01919c09d 100644
--- a/src/armnnTestUtils/CreateWorkload.hpp
+++ b/src/armnnTestUtils/CreateWorkload.hpp
@@ -68,7 +68,7 @@ std::unique_ptr<ActivationWorkload> CreateActivationWorkloadTest(armnn::IWorkloa
{
// Creates the layer we're testing.
ActivationDescriptor layerDesc;
- layerDesc.m_Function = ActivationFunction::Abs;
+ layerDesc.m_Function = ActivationFunction::ReLu;
layerDesc.m_A = 3.5f;
layerDesc.m_B = -10.0f;
@@ -94,7 +94,7 @@ std::unique_ptr<ActivationWorkload> CreateActivationWorkloadTest(armnn::IWorkloa
CHECK(queueDescriptor.m_Outputs.size() == 1);
CHECK(queueDescriptor.m_Parameters.m_A == 3.5f);
CHECK(queueDescriptor.m_Parameters.m_B == -10.0f);
- CHECK((queueDescriptor.m_Parameters.m_Function == ActivationFunction::Abs));
+ CHECK((queueDescriptor.m_Parameters.m_Function == ActivationFunction::ReLu));
// Returns so we can do extra, backend-specific tests.
return workload;