diff options
Diffstat (limited to 'src/backends')
-rw-r--r-- | src/backends/backendsCommon/WorkloadData.cpp | 8 |
1 files changed, 7 insertions, 1 deletions
diff --git a/src/backends/backendsCommon/WorkloadData.cpp b/src/backends/backendsCommon/WorkloadData.cpp index d5e3638a06..af57fee935 100644 --- a/src/backends/backendsCommon/WorkloadData.cpp +++ b/src/backends/backendsCommon/WorkloadData.cpp @@ -946,6 +946,12 @@ void StridedSliceQueueDescriptor::Validate(const WorkloadInfo& workloadInfo) con const TensorInfo& input = workloadInfo.m_InputTensorInfos[0]; const uint32_t rank = input.GetNumDimensions(); + if (rank > 4) + { + throw InvalidArgumentException( + "StridedSliceLayer: Input tensors with rank greater than 4 are not supported"); + } + // Begin, End & Stride length must be of rank(input0) if (m_Parameters.m_Begin.size() != rank) { @@ -975,4 +981,4 @@ void StridedSliceQueueDescriptor::Validate(const WorkloadInfo& workloadInfo) con } } -} //namespace armnn
\ No newline at end of file +} //namespace armnn |