From c2044fe9d26a8b6afca48aee04bd5d29f8e27b8d Mon Sep 17 00:00:00 2001
From: David Beck <david.beck@arm.com>
Date: Wed, 5 Sep 2018 15:00:38 +0100
Subject: IVGCVSW-1804 : Add Subtraction layer types and placeholders

Change-Id: Ib9a477e5ce590df74ba05fece77258b9204f6523
---
 src/armnn/backends/WorkloadData.cpp | 13 +++++++++++++
 1 file changed, 13 insertions(+)

(limited to 'src/armnn/backends/WorkloadData.cpp')

diff --git a/src/armnn/backends/WorkloadData.cpp b/src/armnn/backends/WorkloadData.cpp
index 660637e96f..e3cf83fc6c 100644
--- a/src/armnn/backends/WorkloadData.cpp
+++ b/src/armnn/backends/WorkloadData.cpp
@@ -811,4 +811,17 @@ void DivisionQueueDescriptor::Validate(const WorkloadInfo& workloadInfo) const
                                        "second input");
 }
 
+void SubtractionQueueDescriptor::Validate(const WorkloadInfo& workloadInfo) const
+{
+    ValidateTwoInputs(workloadInfo, "SubtractionQueueDescriptor");
+    ValidateSingleOutput(workloadInfo, "SubtractionQueueDescriptor");
+
+    ValidateBroadcastTensorShapesMatch(workloadInfo.m_InputTensorInfos[0],
+                                       workloadInfo.m_InputTensorInfos[1],
+                                       workloadInfo.m_OutputTensorInfos[0],
+                                       "SubtractionQueueDescriptor",
+                                       "first input",
+                                       "second input");
+}
+
 } //namespace armnn
-- 
cgit v1.2.1