diff options
| author | Teresa Charlin <teresa.charlinreyes@arm.com> | 2021-02-11 23:05:40 +0000 |
|---|---|---|
| committer | Teresa Charlin <teresa.charlinreyes@arm.com> | 2021-02-11 23:07:30 +0000 |
| commit | 2226ca98e01a8f7de37357ebdb2a0ed14fd3b0d2 (patch) | |
| tree | 4dfe5e1768f023c71f23b9c689a5e37bb8f1ec8e /src/backends/backendsCommon/WorkloadData.cpp | |
| parent | 49bdb794170c3d25e3e51fc7b4c267c3d8dbcebf (diff) | |
| download | armnn-2226ca98e01a8f7de37357ebdb2a0ed14fd3b0d2.tar.gz | |
MLCE-347 Bug fixes in Reduce: QueueDescriptor.validate and init REDUCE_MIN
* Allow input tensors of any rank in ReduceQueueDescriptor::validate
* Fix VTS tests failing for REDUCE_MIN due to initialization
Signed-off-by: Teresa Charlin <teresa.charlinreyes@arm.com>
Change-Id: Id8fba1662ade4e0a967093fe5a53b275847f2393
Diffstat (limited to 'src/backends/backendsCommon/WorkloadData.cpp')
| -rw-r--r-- | src/backends/backendsCommon/WorkloadData.cpp | 2 |
1 files changed, 0 insertions, 2 deletions
diff --git a/src/backends/backendsCommon/WorkloadData.cpp b/src/backends/backendsCommon/WorkloadData.cpp index b51099ff79..90db57f953 100644 --- a/src/backends/backendsCommon/WorkloadData.cpp +++ b/src/backends/backendsCommon/WorkloadData.cpp @@ -3643,8 +3643,6 @@ void ReduceQueueDescriptor::Validate(const WorkloadInfo& workloadInfo) const const TensorInfo& inputTensorInfo = workloadInfo.m_InputTensorInfos[0]; const TensorInfo& outputTensorInfo = workloadInfo.m_OutputTensorInfos[0]; - ValidateTensorNumDimensions(inputTensorInfo, descriptorName, 4, "input"); - std::vector<DataType> supportedTypes = { DataType::BFloat16, |