diff options
| author | surmeh01 <surabhi.mehta@arm.com> | 2018-03-29 16:29:27 +0100 |
|---|---|---|
| committer | surmeh01 <surabhi.mehta@arm.com> | 2018-03-29 16:29:27 +0100 |
| commit | bceff2fb3fc68bb0aa88b886900c34b77340c826 (patch) | |
| tree | d867d3e090d58d3012dfbbac456e9ea8c7f789bc /src/armnn/backends/WorkloadData.cpp | |
| parent | 4fcda0101ec3d110c1d6d7bee5c83416b645528a (diff) | |
| download | armnn-bceff2fb3fc68bb0aa88b886900c34b77340c826.tar.gz | |
Release 18.03
Diffstat (limited to 'src/armnn/backends/WorkloadData.cpp')
| -rw-r--r-- | src/armnn/backends/WorkloadData.cpp | 17 |
1 files changed, 7 insertions, 10 deletions
diff --git a/src/armnn/backends/WorkloadData.cpp b/src/armnn/backends/WorkloadData.cpp index 96a37802f1..c951fc5d8d 100644 --- a/src/armnn/backends/WorkloadData.cpp +++ b/src/armnn/backends/WorkloadData.cpp @@ -502,16 +502,13 @@ void MultiplicationQueueDescriptor::Validate(const WorkloadInfo& workloadInfo) c { ValidateTwoInputs(workloadInfo, "MultiplicationQueueDescriptor"); ValidateSingleOutput(workloadInfo, "MultiplicationQueueDescriptor"); - ValidateTensorShapesMatch(workloadInfo.m_InputTensorInfos[0], - workloadInfo.m_InputTensorInfos[1], - "MultiplicationQueueDescriptor", - "first input", - "second input"); - ValidateTensorShapesMatch(workloadInfo.m_InputTensorInfos[0], - workloadInfo.m_OutputTensorInfos[0], - "MultiplicationQueueDescriptor", - "input", - "output"); + + ValidateBroadcastTensorShapesMatch(workloadInfo.m_InputTensorInfos[0], + workloadInfo.m_InputTensorInfos[1], + workloadInfo.m_OutputTensorInfos[0], + "MultiplicationQueueDescriptor", + "first input", + "second input"); } void BatchNormalizationQueueDescriptor::Validate(const WorkloadInfo& workloadInfo) const |