diff options
author | jimfly01 <jim.flynn@arm.com> | 2018-09-19 10:59:49 +0100 |
---|---|---|
committer | Matthew Bentham <matthew.bentham@arm.com> | 2018-10-10 16:16:56 +0100 |
commit | 2c9322a275b0450fd8a7c49b54924890639d2001 (patch) | |
tree | f044af9d97bb5b08c643ddaeba67b48177186152 /src/armnn/backends/WorkloadData.cpp | |
parent | 0cff16365b2ebb4d02929b86a304b91df19a985c (diff) | |
download | armnn-2c9322a275b0450fd8a7c49b54924890639d2001.tar.gz |
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
Diffstat (limited to 'src/armnn/backends/WorkloadData.cpp')
-rw-r--r-- | src/armnn/backends/WorkloadData.cpp | 18 |
1 files changed, 18 insertions, 0 deletions
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 |