diff options
| author | Teresa Charlin <teresa.charlinreyes@arm.com> | 2020-08-07 16:00:38 +0100 |
|---|---|---|
| committer | TeresaARM <teresa.charlinreyes@arm.com> | 2020-08-10 10:49:31 +0000 |
| commit | 9cf7f88f5923dd1ba5d159ade85fcc20185c323f (patch) | |
| tree | 245b89961e45a3ee4309fafef681639d1635e2c7 /src/backends/backendsCommon/test/JsonPrinterTestImpl.cpp | |
| parent | 76615a5edd55b890acdd5fb078d9242e1e719a45 (diff) | |
| download | armnn-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/backends/backendsCommon/test/JsonPrinterTestImpl.cpp')
| -rw-r--r-- | src/backends/backendsCommon/test/JsonPrinterTestImpl.cpp | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/src/backends/backendsCommon/test/JsonPrinterTestImpl.cpp b/src/backends/backendsCommon/test/JsonPrinterTestImpl.cpp index 6251507e51..0452919e05 100644 --- a/src/backends/backendsCommon/test/JsonPrinterTestImpl.cpp +++ b/src/backends/backendsCommon/test/JsonPrinterTestImpl.cpp @@ -134,10 +134,10 @@ std::string GetSoftmaxProfilerJson(const std::vector<armnn::BackendId>& backends IConnectableLayer* input = net->AddInputLayer(0, "input"); SoftmaxDescriptor softmaxDescriptor; - // Set Axis to 0 if CL or Neon until further Axes are supported. + // Set Axis to -1 if CL or Neon until further Axes are supported. if ( backends.front() == armnn::Compute::CpuAcc || backends.front() == armnn::Compute::GpuAcc) { - softmaxDescriptor.m_Axis = 0; + softmaxDescriptor.m_Axis = -1; } IConnectableLayer* softmax = net->AddSoftmaxLayer(softmaxDescriptor, "softmax"); IConnectableLayer* output = net->AddOutputLayer(0, "output"); |