diff options
author | Mohamed Nour Abouelseoud <mohamednour.abouelseoud@arm.com> | 2018-12-10 14:19:41 +0000 |
---|---|---|
committer | Mohamed Nour Abouelseoud <mohamednour.abouelseoud@arm.com> | 2018-12-10 15:24:04 +0000 |
commit | cf22ca9d9121666f985eb7d130aac03233902007 (patch) | |
tree | f962cbc9ca9fbd566320fef9e7d67e4b91d47151 /src/backends/neon/workloads/NeonSoftmaxBaseWorkload.cpp | |
parent | 955258dde076a40373ff9044b5f84c095a0aacef (diff) | |
download | armnn-cf22ca9d9121666f985eb7d130aac03233902007.tar.gz |
IVGCVSW-2354 Remove dimension validation from softmax
Change-Id: I0d69a72df9e348cbeca690926485c050499d1051
Diffstat (limited to 'src/backends/neon/workloads/NeonSoftmaxBaseWorkload.cpp')
-rw-r--r-- | src/backends/neon/workloads/NeonSoftmaxBaseWorkload.cpp | 6 |
1 files changed, 0 insertions, 6 deletions
diff --git a/src/backends/neon/workloads/NeonSoftmaxBaseWorkload.cpp b/src/backends/neon/workloads/NeonSoftmaxBaseWorkload.cpp index b97cf84afd..434de874ae 100644 --- a/src/backends/neon/workloads/NeonSoftmaxBaseWorkload.cpp +++ b/src/backends/neon/workloads/NeonSoftmaxBaseWorkload.cpp @@ -14,12 +14,6 @@ arm_compute::Status NeonSoftmaxWorkloadValidate(const TensorInfo& input, const TensorInfo& output, const SoftmaxDescriptor& descriptor) { - // NOTE: We report 4D Softmax as unsupported until full support is added to ACL - if(input.GetShape().GetNumDimensions() >= 4u) - { - return arm_compute::Status(arm_compute::ErrorCode::RUNTIME_ERROR, "4d softmax is not supported"); - } - const arm_compute::TensorInfo aclInputInfo = armcomputetensorutils::BuildArmComputeTensorInfo(input); const arm_compute::TensorInfo aclOutputInfo = armcomputetensorutils::BuildArmComputeTensorInfo(output); |