diff options
author | David Beck <david.beck@arm.com> | 2018-09-19 12:03:20 +0100 |
---|---|---|
committer | Matthew Bentham <matthew.bentham@arm.com> | 2018-10-10 16:16:56 +0100 |
commit | 10b4dfd8e9ccd7a03df7bb053ee1c644cb37f8ab (patch) | |
tree | 1ac5b4f415531e2ef759439ab8e113f177bea7c5 /src/armnn/backends/RefWorkloads/Broadcast.hpp | |
parent | a3f165624b2cdfbced674af5a6e11856b1e746d9 (diff) | |
download | armnn-10b4dfd8e9ccd7a03df7bb053ee1c644cb37f8ab.tar.gz |
IVGCVSW-1897 : build infrastructure for the src/backends folder
Change-Id: I7ebafb675ccc77ad54d1deb01412a8379a5356bb
Diffstat (limited to 'src/armnn/backends/RefWorkloads/Broadcast.hpp')
-rw-r--r-- | src/armnn/backends/RefWorkloads/Broadcast.hpp | 58 |
1 files changed, 0 insertions, 58 deletions
diff --git a/src/armnn/backends/RefWorkloads/Broadcast.hpp b/src/armnn/backends/RefWorkloads/Broadcast.hpp deleted file mode 100644 index e92ed0598d..0000000000 --- a/src/armnn/backends/RefWorkloads/Broadcast.hpp +++ /dev/null @@ -1,58 +0,0 @@ -// -// Copyright © 2017 Arm Ltd. All rights reserved. -// SPDX-License-Identifier: MIT -// - -#include <armnn/Tensor.hpp> - -#include <functional> - -namespace armnn -{ - -struct BroadcastLoop -{ - BroadcastLoop(const TensorShape& inShape0, const TensorShape& inShape1, const TensorShape& outShape); - - unsigned int GetNumDimensions() - { - return static_cast<unsigned int>(m_DimData.size()); - } - - template <typename T0, typename T1, typename U, typename Func> - void Unroll(Func operationFunc, - unsigned int dimension, - const T0* inData0, - const T1* inData1, - U* outData) - { - if (dimension >= GetNumDimensions()) - { - *outData = operationFunc(*inData0, *inData1); - return; - } - - for (unsigned int i = 0; i < m_DimData[dimension].m_DimSize; i++) - { - Unroll(operationFunc, dimension + 1, inData0, inData1, outData); - - inData0 += m_DimData[dimension].m_Stride1; - inData1 += m_DimData[dimension].m_Stride2; - outData += m_DimData[dimension].m_StrideOut; - } - } - -private: - // Struct to hold the dimension data. - struct BroadcastDimensionData - { - unsigned int m_DimSize; - unsigned int m_StrideOut; - unsigned int m_Stride1; - unsigned int m_Stride2; - }; - - std::vector<BroadcastDimensionData> m_DimData; -}; - -} //namespace armnn
\ No newline at end of file |