diff options
author | Nattapat Chaimanowong <nattapat.chaimanowong@arm.com> | 2018-11-23 15:33:41 +0000 |
---|---|---|
committer | nattapat.chaimanowong <nattapat.chaimanowong@arm.com> | 2018-11-26 16:43:23 +0000 |
commit | 1216b585e085bc3aa0941b7dea6e263e978cb22c (patch) | |
tree | bd04eb4cc92383f788acb79489ecce81ef1bfac5 /src/backends/backendsCommon/test/LayerTests.cpp | |
parent | 144c01b56d5e8b2f9d8e84a03f7fe975888ee25a (diff) | |
download | armnn-1216b585e085bc3aa0941b7dea6e263e978cb22c.tar.gz |
IVGCVSW-2087 Reference implementation and unit tests for StridedSlice
Change-Id: Ifeacc0adb4547c72537b9ea7a61bf3c4ec3673fa
Diffstat (limited to 'src/backends/backendsCommon/test/LayerTests.cpp')
-rwxr-xr-x | src/backends/backendsCommon/test/LayerTests.cpp | 127 |
1 files changed, 127 insertions, 0 deletions
diff --git a/src/backends/backendsCommon/test/LayerTests.cpp b/src/backends/backendsCommon/test/LayerTests.cpp index f10d14e942..648bee657d 100755 --- a/src/backends/backendsCommon/test/LayerTests.cpp +++ b/src/backends/backendsCommon/test/LayerTests.cpp @@ -32,6 +32,7 @@ #include "SpaceToBatchNdTestImpl.hpp" #include "SplitterTestImpl.hpp" #include "SoftmaxTestImpl.hpp" +#include "StridedSliceTestImpl.hpp" #include "NormTestImpl.hpp" #include "PermuteTestImpl.hpp" #include "LstmTestImpl.hpp" @@ -7297,3 +7298,129 @@ LayerTestResult<uint8_t, 4> BatchToSpaceNdNhwcUintTest1( return BatchToSpaceNdHelper<uint8_t, 4, 4>(workloadFactory, memoryManager, armnn::DataLayout::NHWC, inputShape, input, blockShape, crops, outputShape, expectedOutput); } + +LayerTestResult<float, 4> StridedSlice4DFloat32Test( + armnn::IWorkloadFactory& workloadFactory, + const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager) +{ + return StridedSlice4DTest<float>(workloadFactory, memoryManager); +} + +LayerTestResult<float, 4> StridedSlice4DReverseFloat32Test( + armnn::IWorkloadFactory& workloadFactory, + const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager) +{ + return StridedSlice4DReverseTest<float>(workloadFactory, memoryManager); +} + +LayerTestResult<float, 4> StridedSliceSimpleStrideFloat32Test( + armnn::IWorkloadFactory& workloadFactory, + const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager) +{ + return StridedSliceSimpleStrideTest<float>(workloadFactory, memoryManager); +} + +LayerTestResult<float, 4> StridedSliceSimpleRangeMaskFloat32Test( + armnn::IWorkloadFactory& workloadFactory, + const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager) +{ + return StridedSliceSimpleRangeMaskTest<float>(workloadFactory, memoryManager); +} + +LayerTestResult<float, 2> StridedSliceShrinkAxisMaskFloat32Test( + armnn::IWorkloadFactory& workloadFactory, + const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager) +{ + return StridedSliceShrinkAxisMaskTest<float>(workloadFactory, memoryManager); +} + +LayerTestResult<float, 3> StridedSlice3DFloat32Test( + armnn::IWorkloadFactory& workloadFactory, + const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager) +{ + return StridedSlice3DTest<float>(workloadFactory, memoryManager); +} + +LayerTestResult<float, 3> StridedSlice3DReverseFloat32Test( + armnn::IWorkloadFactory& workloadFactory, + const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager) +{ + return StridedSlice3DReverseTest<float>(workloadFactory, memoryManager); +} + +LayerTestResult<float, 2> StridedSlice2DFloat32Test( + armnn::IWorkloadFactory& workloadFactory, + const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager) +{ + return StridedSlice2DTest<float>(workloadFactory, memoryManager); +} + +LayerTestResult<float, 2> StridedSlice2DReverseFloat32Test( + armnn::IWorkloadFactory& workloadFactory, + const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager) +{ + return StridedSlice2DReverseTest<float>(workloadFactory, memoryManager); +} + +LayerTestResult<uint8_t, 4> StridedSlice4DUint8Test( + armnn::IWorkloadFactory& workloadFactory, + const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager) +{ + return StridedSlice4DTest<uint8_t>(workloadFactory, memoryManager); +} + +LayerTestResult<uint8_t, 4> StridedSlice4DReverseUint8Test( + armnn::IWorkloadFactory& workloadFactory, + const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager) +{ + return StridedSlice4DReverseTest<uint8_t>(workloadFactory, memoryManager); +} + +LayerTestResult<uint8_t, 4> StridedSliceSimpleStrideUint8Test( + armnn::IWorkloadFactory& workloadFactory, + const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager) +{ + return StridedSliceSimpleStrideTest<uint8_t>(workloadFactory, memoryManager); +} + +LayerTestResult<uint8_t, 4> StridedSliceSimpleRangeMaskUint8Test( + armnn::IWorkloadFactory& workloadFactory, + const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager) +{ + return StridedSliceSimpleRangeMaskTest<uint8_t>(workloadFactory, memoryManager); +} + +LayerTestResult<uint8_t, 2> StridedSliceShrinkAxisMaskUint8Test( + armnn::IWorkloadFactory& workloadFactory, + const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager) +{ + return StridedSliceShrinkAxisMaskTest<uint8_t>(workloadFactory, memoryManager); +} + +LayerTestResult<uint8_t, 3> StridedSlice3DUint8Test( + armnn::IWorkloadFactory& workloadFactory, + const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager) +{ + return StridedSlice3DTest<uint8_t>(workloadFactory, memoryManager); +} + +LayerTestResult<uint8_t, 3> StridedSlice3DReverseUint8Test( + armnn::IWorkloadFactory& workloadFactory, + const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager) +{ + return StridedSlice3DReverseTest<uint8_t>(workloadFactory, memoryManager); +} + +LayerTestResult<uint8_t, 2> StridedSlice2DUint8Test( + armnn::IWorkloadFactory& workloadFactory, + const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager) +{ + return StridedSlice2DTest<uint8_t>(workloadFactory, memoryManager); +} + +LayerTestResult<uint8_t, 2> StridedSlice2DReverseUint8Test( + armnn::IWorkloadFactory& workloadFactory, + const armnn::IBackendInternal::IMemoryManagerSharedPtr& memoryManager) +{ + return StridedSlice2DReverseTest<uint8_t>(workloadFactory, memoryManager); +} |