From d5166106ebfc11b1db832a708892e12edeaf997e Mon Sep 17 00:00:00 2001 From: Matthew Jackson Date: Wed, 31 Jul 2019 14:06:28 +0100 Subject: IVGCVSW-3539 Add CL support and tests for Stack * Added CL backend support for Stack * Added unit tests for Stack on the CL backend * Refactored unit tests to support generic data types Signed-off-by: Matthew Jackson Change-Id: I38ee3e9d8947ea98a3104c982698001e704d7d89 --- src/armnn/test/CreateWorkload.hpp | 9 ++++----- 1 file changed, 4 insertions(+), 5 deletions(-) (limited to 'src/armnn/test') diff --git a/src/armnn/test/CreateWorkload.hpp b/src/armnn/test/CreateWorkload.hpp index 619c6d02cd..98cdfaff0e 100644 --- a/src/armnn/test/CreateWorkload.hpp +++ b/src/armnn/test/CreateWorkload.hpp @@ -1381,17 +1381,16 @@ std::unique_ptr CreateSpaceToDepthWorkloadTest(armnn::IWor return workload; } -template +template std::unique_ptr CreateStackWorkloadTest(armnn::IWorkloadFactory& factory, armnn::Graph& graph, const armnn::TensorShape& inputShape, const armnn::TensorShape& outputShape, unsigned int axis, - unsigned int numInputs, - armnn::DataType dataType) + unsigned int numInputs) { - armnn::TensorInfo inputTensorInfo(inputShape, dataType); - armnn::TensorInfo outputTensorInfo(outputShape, dataType); + armnn::TensorInfo inputTensorInfo(inputShape, DataType); + armnn::TensorInfo outputTensorInfo(outputShape, DataType); // Constructs the Stack layer. armnn::StackDescriptor descriptor(axis, numInputs, inputShape); -- cgit v1.2.1