aboutsummaryrefslogtreecommitdiff
path: root/src/backends/backendsCommon/WorkloadData.cpp
diff options
context:
space:
mode:
Diffstat (limited to 'src/backends/backendsCommon/WorkloadData.cpp')
-rw-r--r--src/backends/backendsCommon/WorkloadData.cpp66
1 files changed, 66 insertions, 0 deletions
diff --git a/src/backends/backendsCommon/WorkloadData.cpp b/src/backends/backendsCommon/WorkloadData.cpp
index d4ae08d874..6cde89c2e1 100644
--- a/src/backends/backendsCommon/WorkloadData.cpp
+++ b/src/backends/backendsCommon/WorkloadData.cpp
@@ -1640,6 +1640,72 @@ void ResizeQueueDescriptor::Validate(const WorkloadInfo& workloadInfo) const
}
}
+void ReverseV2QueueDescriptor::Validate(const WorkloadInfo &workloadInfo) const {
+ const std::string descriptorName{"ReverseV2QueueDescriptor"};
+
+ ValidateNumInputs(workloadInfo, descriptorName, 1);
+ ValidateNumOutputs(workloadInfo, descriptorName, 1);
+
+ const TensorInfo& inputTensorInfo = workloadInfo.m_InputTensorInfos[0];
+ const TensorInfo& outputTensorInfo = workloadInfo.m_OutputTensorInfos[0];
+
+ auto inputTensorNumDimensions = inputTensorInfo.GetNumDimensions();
+ if (inputTensorNumDimensions > m_Parameters.m_MaxDimension)
+ {
+ throw InvalidArgumentException(descriptorName +
+ ": Input tensors with rank greater than " +
+ std::to_string(m_Parameters.m_MaxDimension) + " are not supported.");
+ }
+
+ std::vector<DataType> supportedTypes =
+ {
+ DataType::BFloat16,
+ DataType::Float16,
+ DataType::Float32,
+ DataType::QAsymmS8,
+ DataType::QAsymmU8,
+ DataType::QSymmS16
+ };
+
+ ValidateDataTypes(inputTensorInfo, supportedTypes, descriptorName);
+ ValidateTensorShapesMatch(inputTensorInfo, outputTensorInfo, descriptorName, "input", "output");
+ ValidateTensorDataTypesMatch(inputTensorInfo, outputTensorInfo, descriptorName, "input", "output");
+
+ if (m_Parameters.m_Axis.size() > inputTensorNumDimensions)
+ {
+ throw InvalidArgumentException(descriptorName + ": More axes specified than is on the input tensor.");
+ }
+ if (m_Parameters.m_Axis.size() > m_Parameters.m_MaxDimension)
+ {
+ throw InvalidArgumentException(descriptorName +
+ ": More than " + std::to_string(m_Parameters.m_MaxDimension) + " axes cannot be specified.");
+ }
+
+ if (! m_Parameters.m_Axis.empty())
+ {
+ // First check that we have unique axis values
+ auto checkAxis = m_Parameters.m_Axis;
+ std::sort(checkAxis.begin(), checkAxis.end());
+ auto lastUnique = std::unique(checkAxis.begin(), checkAxis.end());
+ if (lastUnique != checkAxis.end())
+ {
+ throw InvalidArgumentException(descriptorName + ": Axes values must be unique.");
+ }
+
+ // Next check that the axes values are in range: [-rank, rank]
+ const auto minmax =
+ std::minmax_element(std::begin(m_Parameters.m_Axis), std::end(m_Parameters.m_Axis));
+ if (((*minmax.first) < int32_t(-inputTensorNumDimensions)) ||
+ ((*minmax.second) >= int32_t (inputTensorNumDimensions)))
+ {
+ throw InvalidArgumentException(descriptorName +
+ ": Axes values must in range [-" + std::to_string(inputTensorNumDimensions) + "," +
+ std::to_string(inputTensorNumDimensions) + "].");
+ }
+ }
+}
+
+
void FakeQuantizationQueueDescriptor::Validate(const WorkloadInfo& workloadInfo) const
{
const std::string descriptorName{"FakeQuantizationQueueDescriptor"};