diff options
author | Tianle Cheng <tianle.cheng@arm.com> | 2023-06-28 13:20:47 +0100 |
---|---|---|
committer | Tianle Cheng <tianle.cheng@arm.com> | 2023-07-04 10:36:43 +0000 |
commit | 988354de127528bdebb98fd25661fbf2f39f17dd (patch) | |
tree | c06f5250bdd0182055ac9e84e20d6e338518ad08 /src/backends/backendsCommon/WorkloadData.cpp | |
parent | 9414936e62ed8cd18cc33c0390bb605a782556c6 (diff) | |
download | armnn-988354de127528bdebb98fd25661fbf2f39f17dd.tar.gz |
IVGCVSW-7831: Front end and Reference Implementation for REVERSE_V2
* Descriptors added for ReverseV2
* Layer definition added
* Input validation added
* Reference workload implementation for ReverseV2 added
* Reference layer unit tests made for ReverseV2
* CompareTensors method updated to support comparison between empty tensors
* CMake and other build files updated
Signed-off-by: Tianle Cheng <tianle.cheng@arm.com>
Change-Id: I805738454421309fda77c44218a8df171d68dc18
Diffstat (limited to 'src/backends/backendsCommon/WorkloadData.cpp')
-rw-r--r-- | src/backends/backendsCommon/WorkloadData.cpp | 66 |
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"}; |