diff options
author | Nattapat Chaimanowong <nattapat.chaimanowong@arm.com> | 2018-11-21 16:48:17 +0000 |
---|---|---|
committer | nattapat.chaimanowong <nattapat.chaimanowong@arm.com> | 2018-11-21 17:13:30 +0000 |
commit | a0d2844d05dd9ae15733d426b04ab651457604ce (patch) | |
tree | 7ae0787d7be49ec2b1bb754ff35e13e8b71c73d2 /src/backends/backendsCommon/WorkloadData.cpp | |
parent | 3d93bc47f42b339d82cfcf56a90c9264dd46d70a (diff) | |
download | armnn-a0d2844d05dd9ae15733d426b04ab651457604ce.tar.gz |
IVGCVSW-2086 Update StridedSliceLayer and StridedSliceDescriptor
Change-Id: Ifa88a879dd239f60ab27330d6b73859393828ef0
Diffstat (limited to 'src/backends/backendsCommon/WorkloadData.cpp')
-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 |