diff options
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 |