From d41b25938323455ea6b6d5348cab8861971b5fba Mon Sep 17 00:00:00 2001 From: Nina Drozd Date: Mon, 19 Nov 2018 13:03:36 +0000 Subject: IVGCVSW-2144: Adding TensorUtils class * helper methods for creating TensorShape and TensorInfo objects Change-Id: I371fc7aea08ca6bbb9c205a143ce36e8353a1c48 --- src/armnn/test/TensorHelpers.hpp | 19 ------------------- 1 file changed, 19 deletions(-) (limited to 'src/armnn') diff --git a/src/armnn/test/TensorHelpers.hpp b/src/armnn/test/TensorHelpers.hpp index 238232ae96..06818d3918 100644 --- a/src/armnn/test/TensorHelpers.hpp +++ b/src/armnn/test/TensorHelpers.hpp @@ -210,22 +210,3 @@ boost::multi_array MakeRandomTensor(const armnn::TensorInfo& tensorInfo, int32_t qOffset = tensorInfo.GetQuantizationOffset(); return MakeTensor(tensorInfo, QuantizedVector(qScale, qOffset, init)); } - -template -armnn::TensorInfo GetTensorInfo(unsigned int numberOfBatches, - unsigned int numberOfChannels, - unsigned int height, - unsigned int width, - const armnn::DataLayout dataLayout) -{ - switch (dataLayout) - { - case armnn::DataLayout::NCHW: - return armnn::TensorInfo({numberOfBatches, numberOfChannels, height, width}, armnn::GetDataType()); - case armnn::DataLayout::NHWC: - return armnn::TensorInfo({numberOfBatches, height, width, numberOfChannels}, armnn::GetDataType()); - default: - throw armnn::InvalidArgumentException("unknown data layout [" - + std::to_string(static_cast(dataLayout)) + "]"); - } -} -- cgit v1.2.1