From 2c9322a275b0450fd8a7c49b54924890639d2001 Mon Sep 17 00:00:00 2001 From: jimfly01 Date: Wed, 19 Sep 2018 10:59:49 +0100 Subject: IVGCVSW-1880: Add no-op factory implementations of CreatePad * Added PadDescriptor and PadQueueDescriptor * Factory method CreatePad on all backends returning null workloads Change-Id: Ifeb50f9249ea31edfc3586d49b008ea1b5c59ae2 --- src/armnn/backends/WorkloadData.cpp | 18 ++++++++++++++++++ 1 file changed, 18 insertions(+) (limited to 'src/armnn/backends/WorkloadData.cpp') diff --git a/src/armnn/backends/WorkloadData.cpp b/src/armnn/backends/WorkloadData.cpp index 25144a4753..c5c607d954 100644 --- a/src/armnn/backends/WorkloadData.cpp +++ b/src/armnn/backends/WorkloadData.cpp @@ -850,4 +850,22 @@ void MeanQueueDescriptor::Validate(const WorkloadInfo& workloadInfo) const } } +void PadQueueDescriptor::Validate(const WorkloadInfo& workloadInfo) const +{ + ValidateSingleInput(workloadInfo, "PadQueueDescriptor"); + ValidateSingleOutput(workloadInfo, "PadQueueDescriptor"); + + const TensorInfo& input = workloadInfo.m_InputTensorInfos[0]; + const TensorInfo& output = workloadInfo.m_OutputTensorInfos[1]; + // input and output should have the same number of dimensions + ValidateTensorNumDimensions(output, "PadQueueDescriptor", input.GetNumDimensions(), "output"); + // there should be entry in the pad list for each dimension in the input tensor + if (m_Parameters.m_PadList.size() != input.GetNumDimensions()) { + throw InvalidArgumentException("Pad List should contain the same number of entries as there" + " are dimensions in the input tensor that is " + + to_string(input.GetNumDimensions()) + " entries " + + " not " + to_string(m_Parameters.m_PadList.size()) + " entries."); + } +} + } //namespace armnn -- cgit v1.2.1