From 5e90aab1cc25681c3e02b4d4436c24ee43400e91 Mon Sep 17 00:00:00 2001 From: Georgios Pinitas Date: Fri, 14 Feb 2020 14:46:51 +0000 Subject: COMPMID-3059: Add TF parser support for StridedSlice Signed-off-by: Georgios Pinitas Change-Id: I31f25f26a50c9054b5650b1be127c84194b56be7 --- src/armnnUtils/ParserHelper.hpp | 5 +++++ 1 file changed, 5 insertions(+) (limited to 'src/armnnUtils/ParserHelper.hpp') diff --git a/src/armnnUtils/ParserHelper.hpp b/src/armnnUtils/ParserHelper.hpp index d63408804d..28c7964ac1 100644 --- a/src/armnnUtils/ParserHelper.hpp +++ b/src/armnnUtils/ParserHelper.hpp @@ -25,4 +25,9 @@ void CalculateReducedOutputTensoInfo(const armnn::TensorInfo& inputTensorInfo, bool keepDims, armnn::TensorInfo& outputTensorInfo); +/// Create output tensor info for a StridedSlice operator +void CalculateStridedSliceOutputTensorInfo(const armnn::TensorInfo& inputTensorInfo, + const armnn::StridedSliceDescriptor& desc, + armnn::TensorInfo& outputTensorInfo); + } // namespace armnnUtils -- cgit v1.2.1