diff options
author | Nina Drozd <nina.drozd@arm.com> | 2018-10-02 11:14:17 +0100 |
---|---|---|
committer | Matthew Bentham <matthew.bentham@arm.com> | 2018-10-10 16:16:58 +0100 |
commit | 661dfa73bd5da4816a42acbb0768d0766d985e1c (patch) | |
tree | f1a874949930b1ecdea21419ce4e40a0bd3564ae /src/backends/WorkloadData.cpp | |
parent | d59116ecb54c5bfe828d82ea0bc3367bc9b8c5dd (diff) | |
download | armnn-661dfa73bd5da4816a42acbb0768d0766d985e1c.tar.gz |
IVGCVSW-1886 - forward layer support for pad
Ref returning false for now until implementation complete in 1885
Change-Id: I28cbc584eaa6ce3ad79a531db7155b3f9716f30c
Diffstat (limited to 'src/backends/WorkloadData.cpp')
-rw-r--r-- | src/backends/WorkloadData.cpp | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/src/backends/WorkloadData.cpp b/src/backends/WorkloadData.cpp index 32ed97a052..a6a87f3782 100644 --- a/src/backends/WorkloadData.cpp +++ b/src/backends/WorkloadData.cpp @@ -858,7 +858,8 @@ void PadQueueDescriptor::Validate(const WorkloadInfo& workloadInfo) const ValidateSingleOutput(workloadInfo, "PadQueueDescriptor"); const TensorInfo& input = workloadInfo.m_InputTensorInfos[0]; - const TensorInfo& output = workloadInfo.m_OutputTensorInfos[1]; + const TensorInfo& output = workloadInfo.m_OutputTensorInfos[0]; + // 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 |