diff options
| author | Nina Drozd <nina.drozd@arm.com> | 2018-11-19 13:03:36 +0000 |
|---|---|---|
| committer | Aron Virginas-Tar <aron.virginas-tar@arm.com> | 2018-11-20 13:04:34 +0000 |
| commit | d41b25938323455ea6b6d5348cab8861971b5fba (patch) | |
| tree | 393d8cb22b32b0b72f23a110d56aa8e3ee830cd8 /src/armnn/test/TensorHelpers.hpp | |
| parent | 63b08829cbb0a36f4ed1f71717626ff38ff909ba (diff) | |
| download | armnn-d41b25938323455ea6b6d5348cab8861971b5fba.tar.gz | |
IVGCVSW-2144: Adding TensorUtils class
* helper methods for creating TensorShape and TensorInfo objects
Change-Id: I371fc7aea08ca6bbb9c205a143ce36e8353a1c48
Diffstat (limited to 'src/armnn/test/TensorHelpers.hpp')
| -rw-r--r-- | src/armnn/test/TensorHelpers.hpp | 19 |
1 files changed, 0 insertions, 19 deletions
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<T, n> MakeRandomTensor(const armnn::TensorInfo& tensorInfo, int32_t qOffset = tensorInfo.GetQuantizationOffset(); return MakeTensor<T, n>(tensorInfo, QuantizedVector<T>(qScale, qOffset, init)); } - -template<typename T> -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<T>()); - case armnn::DataLayout::NHWC: - return armnn::TensorInfo({numberOfBatches, height, width, numberOfChannels}, armnn::GetDataType<T>()); - default: - throw armnn::InvalidArgumentException("unknown data layout [" - + std::to_string(static_cast<int>(dataLayout)) + "]"); - } -} |