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/cl/workloads/ClSoftmaxBaseWorkload.cpp | |
| parent | 955258dde076a40373ff9044b5f84c095a0aacef (diff) | |
| download | armnn-cf22ca9d9121666f985eb7d130aac03233902007.tar.gz | |
IVGCVSW-2354 Remove dimension validation from softmax
Change-Id: I0d69a72df9e348cbeca690926485c050499d1051
Diffstat (limited to 'src/backends/cl/workloads/ClSoftmaxBaseWorkload.cpp')
| -rw-r--r-- | src/backends/cl/workloads/ClSoftmaxBaseWorkload.cpp | 6 |
1 files changed, 0 insertions, 6 deletions
diff --git a/src/backends/cl/workloads/ClSoftmaxBaseWorkload.cpp b/src/backends/cl/workloads/ClSoftmaxBaseWorkload.cpp index 48dd3984f1..b1dc404a6f 100644 --- a/src/backends/cl/workloads/ClSoftmaxBaseWorkload.cpp +++ b/src/backends/cl/workloads/ClSoftmaxBaseWorkload.cpp @@ -15,12 +15,6 @@ namespace armnn arm_compute::Status ClSoftmaxWorkloadValidate(const TensorInfo& input, const TensorInfo& output) { - // 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); |