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/OutputHandler.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/OutputHandler.hpp')
-rw-r--r-- | src/armnn/backends/OutputHandler.hpp | 63 |
1 files changed, 0 insertions, 63 deletions
diff --git a/src/armnn/backends/OutputHandler.hpp b/src/armnn/backends/OutputHandler.hpp deleted file mode 100644 index dfc01844c9..0000000000 --- a/src/armnn/backends/OutputHandler.hpp +++ /dev/null @@ -1,63 +0,0 @@ -// -// Copyright © 2017 Arm Ltd. All rights reserved. -// SPDX-License-Identifier: MIT -// -#pragma once - -#include "backends/WorkloadDataFwd.hpp" - -#include <string> -#include <vector> - -#include <memory> -#include <set> - -#include <boost/assert.hpp> - -#include "armnn/INetwork.hpp" -#include "armnn/Types.hpp" -#include "armnn/Descriptors.hpp" -#include "armnn/Tensor.hpp" -#include "ITensorHandle.hpp" - -namespace armnn -{ - -class ITensorHandle; -class IWorkloadFactory; -class OutputSlot; -class WorkloadDataCollector; - -class OutputHandler -{ -public: - /// @brief - Sets the TensorInfo used by this output handler. - /// @param tensorInfo - TensorInfo for the output. - void SetTensorInfo(const TensorInfo& tensorInfo); - - /// @brief - Creates tensor handlers used by the intermediate tensors. Does not allocate memory. - /// @param factory - Factory to be used for handler creation. - void CreateTensorHandles(const IWorkloadFactory& factory); - - /// @brief - Gets the matching TensorInfo for the output. - /// @return - References to the output TensorInfo. - const TensorInfo& GetTensorInfo() const { return m_TensorInfo; } - - /// @brief - Gets the allocated tensor memory. - /// @return - Pointer to the tensor memory. - ITensorHandle* GetData() const { return m_TensorHandle.get(); } - - /// Fill the outputs for a given queue descriptor. - void CollectWorkloadOutputs(WorkloadDataCollector& dataCollector) const; - - void SetData(std::unique_ptr<ITensorHandle> data) { m_TensorHandle = std::move(data); } - - /// @brief Returns true if SetTensorInfo() has been called at least once on this. - bool IsTensorInfoSet() const { return m_bTensorInfoSet; } -private: - std::unique_ptr<ITensorHandle> m_TensorHandle; - TensorInfo m_TensorInfo; - bool m_bTensorInfoSet = false; -}; - -} //namespace armnn |