From 1216b585e085bc3aa0941b7dea6e263e978cb22c Mon Sep 17 00:00:00 2001 From: Nattapat Chaimanowong Date: Fri, 23 Nov 2018 15:33:41 +0000 Subject: IVGCVSW-2087 Reference implementation and unit tests for StridedSlice Change-Id: Ifeacc0adb4547c72537b9ea7a61bf3c4ec3673fa --- src/backends/reference/workloads/StridedSlice.hpp | 21 +++++++++++++++++++++ 1 file changed, 21 insertions(+) create mode 100644 src/backends/reference/workloads/StridedSlice.hpp (limited to 'src/backends/reference/workloads/StridedSlice.hpp') diff --git a/src/backends/reference/workloads/StridedSlice.hpp b/src/backends/reference/workloads/StridedSlice.hpp new file mode 100644 index 0000000000..8eed8706dc --- /dev/null +++ b/src/backends/reference/workloads/StridedSlice.hpp @@ -0,0 +1,21 @@ +// +// Copyright © 2017 Arm Ltd. All rights reserved. +// SPDX-License-Identifier: MIT +// + +#pragma once + +#include +#include + +namespace armnn +{ + +template +void StridedSlice(const TensorInfo& inputInfo, + const TensorInfo& outputInfo, + const StridedSliceDescriptor& params, + const T* inputData, + T* outputData); + +} //namespace armnn -- cgit v1.2.1