aboutsummaryrefslogtreecommitdiff
path: root/src/armnn/test/CreateWorkload.hpp
diff options
context:
space:
mode:
authorTeresa Charlin <teresa.charlinreyes@arm.com>2020-08-07 16:00:38 +0100
committerTeresaARM <teresa.charlinreyes@arm.com>2020-08-10 10:49:31 +0000
commit9cf7f88f5923dd1ba5d159ade85fcc20185c323f (patch)
tree245b89961e45a3ee4309fafef681639d1635e2c7 /src/armnn/test/CreateWorkload.hpp
parent76615a5edd55b890acdd5fb078d9242e1e719a45 (diff)
downloadarmnn-9cf7f88f5923dd1ba5d159ade85fcc20185c323f.tar.gz
Update ACL pin to 4aed4aafa2ddb0b6f4b76aef5008c8bb45599ea4
* Use axis=-1 in all softmax tests as ACL only accepts this dimension (the most right dimension, in ACL is 0 and in ArmNN is -1). Signed-off-by: Teresa Charlin <teresa.charlinreyes@arm.com> Change-Id: I4be83f60aa9505b4bf2367c4489e6f12d644c6d4
Diffstat (limited to 'src/armnn/test/CreateWorkload.hpp')
-rw-r--r--src/armnn/test/CreateWorkload.hpp8
1 files changed, 4 insertions, 4 deletions
diff --git a/src/armnn/test/CreateWorkload.hpp b/src/armnn/test/CreateWorkload.hpp
index aad6244c4b..fe73550bbc 100644
--- a/src/armnn/test/CreateWorkload.hpp
+++ b/src/armnn/test/CreateWorkload.hpp
@@ -947,10 +947,10 @@ std::unique_ptr<SoftmaxWorkload> CreateSoftmaxWorkloadTest(armnn::IWorkloadFacto
{
// Create the layer we're testing.
SoftmaxDescriptor softmaxDescriptor;
- // Set Axis to 1 if CL or Neon until further Axes are supported.
+ // Set Axis to -1 if CL or Neon until further Axes are supported.
if (factory.GetBackendId() == armnn::Compute::CpuAcc || factory.GetBackendId() == armnn::Compute::GpuAcc)
{
- softmaxDescriptor.m_Axis = 1;
+ softmaxDescriptor.m_Axis = -1;
}
Layer* const layer = graph.AddLayer<SoftmaxLayer>(softmaxDescriptor, "layer");
@@ -1268,10 +1268,10 @@ std::unique_ptr<LogSoftmaxWorkload> CreateLogSoftmaxWorkloadTest(armnn::IWorkloa
{
// Create the layer we're testing.
LogSoftmaxDescriptor logSoftmaxDescriptor;
- // Set Axis to 1 if CL or Neon until further Axes are supported.
+ // Set Axis to -1 if CL or Neon until further Axes are supported.
if (factory.GetBackendId() == armnn::Compute::CpuAcc || factory.GetBackendId() == armnn::Compute::GpuAcc)
{
- logSoftmaxDescriptor.m_Axis = 0;
+ logSoftmaxDescriptor.m_Axis = -1;
}
Layer* const layer = graph.AddLayer<LogSoftmaxLayer>(logSoftmaxDescriptor, "layer");