aboutsummaryrefslogtreecommitdiff
path: root/src/backends/OutputHandler.hpp
diff options
context:
space:
mode:
authorAron Virginas-Tar <Aron.Virginas-Tar@arm.com>2018-11-01 16:15:57 +0000
committerAron Virginas-Tar <Aron.Virginas-Tar@arm.com>2018-11-02 14:49:21 +0000
commitc9cc80455ff29fd2c8622c9487ec9c57ade6ea30 (patch)
tree41b1491312fe6082b39d5d37ffa0dcf0ab0f2817 /src/backends/OutputHandler.hpp
parent207ef9a6b8b3ea0afe9a095639f67b5dedd095d7 (diff)
downloadarmnn-c9cc80455ff29fd2c8622c9487ec9c57ade6ea30.tar.gz
IVGCVSW-1946: Remove armnn/src from the include paths
Change-Id: I663a0a0fccb43ee960ec070121a59df9db0bb04e
Diffstat (limited to 'src/backends/OutputHandler.hpp')
-rw-r--r--src/backends/OutputHandler.hpp68
1 files changed, 0 insertions, 68 deletions
diff --git a/src/backends/OutputHandler.hpp b/src/backends/OutputHandler.hpp
deleted file mode 100644
index ad4a2931cd..0000000000
--- a/src/backends/OutputHandler.hpp
+++ /dev/null
@@ -1,68 +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 - Creates tensor handlers used by the intermediate tensors. Does not allocate memory.
- /// @param factory - Factory to be used for handler creation.
- /// @param dataLayout - Data Layout to be used for handler creation.
- void CreateTensorHandles(const IWorkloadFactory& factory, DataLayout dataLayout);
-
- /// @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