IVGCVSW-2467 Remove GetDataType<T> function

[platform/upstream/armnn.git] / src / armnnUtils / TensorUtils.hpp

diff --git a/src/armnnUtils/TensorUtils.hpp b/src/armnnUtils/TensorUtils.hpp
index 6461b37..fb5e6eb 100644 (file)
--- a/src/armnnUtils/TensorUtils.hpp
+++ b/src/armnnUtils/TensorUtils.hpp
@@ -15,23 +15,11 @@ armnn::TensorShape GetTensorShape(unsigned int numberOfBatches,
                                   unsigned int width,
                                   const armnn::DataLayout dataLayout);
 
-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)) +
-                                                  "]", CHECK_LOCATION());
-    }
-}
-} // namespace armnnUtils
\ No newline at end of file
+                                const armnn::DataLayout dataLayout,
+                                const armnn::DataType dataType);
+
+} // namespace armnnUtils