diff options
author | Matteo Martincigh <matteo.martincigh@arm.com> | 2019-11-28 15:45:42 +0000 |
---|---|---|
committer | Narumol Prangnawarat <narumol.prangnawarat@arm.com> | 2019-12-09 12:00:04 +0000 |
commit | e5b8eb9fe8147a0849db08ef0898a0e8bef920b4 (patch) | |
tree | 54f039796753b6a395eb9f76e46e11a9413dabad /src/backends/backendsCommon/OutputHandler.hpp | |
parent | 3e2969d7195d77796774101580b837681505904a (diff) | |
download | armnn-e5b8eb9fe8147a0849db08ef0898a0e8bef920b4.tar.gz |
IVGCVSW-4210 Create a public API for the common backend files
* Create a public API for the common backend files
* Move OutputHandler to armnn internal
* Remove unused headers
Signed-off-by: Matteo Martincigh <matteo.martincigh@arm.com>
Signed-off-by: Narumol Prangnawarat <narumol.prangnawarat@arm.com>
Change-Id: I3e86d908b021e3561befa9d45158d87d2cbb18c0
Diffstat (limited to 'src/backends/backendsCommon/OutputHandler.hpp')
-rw-r--r-- | src/backends/backendsCommon/OutputHandler.hpp | 63 |
1 files changed, 0 insertions, 63 deletions
diff --git a/src/backends/backendsCommon/OutputHandler.hpp b/src/backends/backendsCommon/OutputHandler.hpp deleted file mode 100644 index 81768c6f5e..0000000000 --- a/src/backends/backendsCommon/OutputHandler.hpp +++ /dev/null @@ -1,63 +0,0 @@ -// -// Copyright © 2017 Arm Ltd. All rights reserved. -// SPDX-License-Identifier: MIT -// -#pragma once - -#include "ITensorHandle.hpp" -#include "ITensorHandleFactory.hpp" - -#include <armnn/Descriptors.hpp> -#include <armnn/INetwork.hpp> -#include <armnn/Tensor.hpp> -#include <armnn/Types.hpp> - -#include <memory> -#include <set> -#include <string> -#include <vector> - -#include <boost/assert.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 handles used by the intermediate tensors. Does not allocate memory. - /// @param factory - Factory to be used for handler creation. - void CreateTensorHandles(const IWorkloadFactory& factory, const bool IsMemoryManaged = true); - void CreateTensorHandles(const ITensorHandleFactory& factory, const bool IsMemoryManaged = true); - - /// @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 |