diff options
Diffstat (limited to 'src/backends/backendsCommon/WorkloadData.cpp')
| -rw-r--r-- | src/backends/backendsCommon/WorkloadData.cpp | 10 |
1 files changed, 10 insertions, 0 deletions
diff --git a/src/backends/backendsCommon/WorkloadData.cpp b/src/backends/backendsCommon/WorkloadData.cpp index 05f4e317a9..9714b02a80 100644 --- a/src/backends/backendsCommon/WorkloadData.cpp +++ b/src/backends/backendsCommon/WorkloadData.cpp @@ -1025,6 +1025,11 @@ void EqualQueueDescriptor::Validate(const WorkloadInfo& workloadInfo) const "EqualQueueDescriptor", "first input", "second input"); + + if (workloadInfo.m_OutputTensorInfos[0].GetDataType() != DataType::Boolean) + { + throw InvalidArgumentException("EqualQueueDescriptor: Output tensor type must be Boolean."); + } } void GreaterQueueDescriptor::Validate(const WorkloadInfo& workloadInfo) const @@ -1038,6 +1043,11 @@ void GreaterQueueDescriptor::Validate(const WorkloadInfo& workloadInfo) const "GreaterQueueDescriptor", "first input", "second input"); + + if (workloadInfo.m_OutputTensorInfos[0].GetDataType() != DataType::Boolean) + { + throw InvalidArgumentException("GreaterQueueDescriptor: Output tensor type must be Boolean."); + } } void RsqrtQueueDescriptor::Validate(const WorkloadInfo& workloadInfo) const |