aboutsummaryrefslogtreecommitdiff
path: root/src/backends/backendsCommon/OutputHandler.hpp
diff options
context:
space:
mode:
authorMatteo Martincigh <matteo.martincigh@arm.com>2019-11-28 15:45:42 +0000
committerNarumol Prangnawarat <narumol.prangnawarat@arm.com>2019-12-09 12:00:04 +0000
commite5b8eb9fe8147a0849db08ef0898a0e8bef920b4 (patch)
tree54f039796753b6a395eb9f76e46e11a9413dabad /src/backends/backendsCommon/OutputHandler.hpp
parent3e2969d7195d77796774101580b837681505904a (diff)
downloadarmnn-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.hpp63
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