From a0d2844d05dd9ae15733d426b04ab651457604ce Mon Sep 17 00:00:00 2001 From: Nattapat Chaimanowong Date: Wed, 21 Nov 2018 16:48:17 +0000 Subject: IVGCVSW-2086 Update StridedSliceLayer and StridedSliceDescriptor Change-Id: Ifa88a879dd239f60ab27330d6b73859393828ef0 --- src/backends/backendsCommon/WorkloadData.cpp | 8 +++++++- 1 file changed, 7 insertions(+), 1 deletion(-) (limited to 'src/backends/backendsCommon/WorkloadData.cpp') 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 -- cgit v1.2.1