projects / platform / upstream / armnn.git / commitdiff
? search:
summary | shortlog | log | commit | commitdiff | tree
raw | patch | inline | side by side (parent: cfdcadf)
IVGCVSW-836 Remove 2D validation in Softmax
authorMohamed Nour Abouelseoud <mohamednour.abouelseoud@arm.com>
Thu, 6 Dec 2018 16:48:39 +0000 (16:48 +0000)
committerLes Bell <les.bell@arm.com>
Fri, 7 Dec 2018 08:51:54 +0000 (08:51 +0000)
Change-Id: I87c53c78dd033871dbc543550a61d2d535e752ac

src/backends/backendsCommon/WorkloadData.cpp patch | blob | history

diff --git a/src/backends/backendsCommon/WorkloadData.cpp b/src/backends/backendsCommon/WorkloadData.cpp
index e0ba4a7..bc25415 100644 (file)
--- a/src/backends/backendsCommon/WorkloadData.cpp
+++ b/src/backends/backendsCommon/WorkloadData.cpp
@@ -313,8 +313,6 @@ void SoftmaxQueueDescriptor::Validate(const WorkloadInfo& workloadInfo) const
 {
     ValidateSingleInput(workloadInfo, "SoftmaxQueueDescriptor");
     ValidateSingleOutput(workloadInfo, "SoftmaxQueueDescriptor");
-    ValidateTensorNumDimensions(workloadInfo.m_InputTensorInfos[0], "SoftmaxQueueDescriptor", 2, "input");
-    ValidateTensorNumDimensions(workloadInfo.m_OutputTensorInfos[0], "SoftmaxQueueDescriptor", 2, "output");
 
     ValidateTensorShapesMatch(workloadInfo.m_InputTensorInfos[0],
                               workloadInfo.m_OutputTensorInfos[0],
Domain: Machine Learning / ML Framework;