aboutsummaryrefslogtreecommitdiff
path: root/src/armnn/backends/OutputHandler.hpp
diff options
context:
space:
mode:
Diffstat (limited to 'src/armnn/backends/OutputHandler.hpp')
-rw-r--r--src/armnn/backends/OutputHandler.hpp21
1 files changed, 9 insertions, 12 deletions
diff --git a/src/armnn/backends/OutputHandler.hpp b/src/armnn/backends/OutputHandler.hpp
index 9cc87c6095..ed95577cca 100644
--- a/src/armnn/backends/OutputHandler.hpp
+++ b/src/armnn/backends/OutputHandler.hpp
@@ -31,30 +31,27 @@ class WorkloadDataCollector;
class OutputHandler
{
public:
- /// @brief Sets the TensorInfo used by this output handler.
- /// @param tensorInfo TensorInfo for the output.
+ /// @brief - Sets the TensorInfo used by this output handler.
+ /// @param tensorInfo - TensorInfo for the output.
void SetTensorInfo(const TensorInfo& tensorInfo);
- /// @brief Create tensor handlers used by the intermediate tensors. Does not allocate memory.
- /// @param factory Factory to be used for handler creation.
+ /// @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 Get the matching TensorInfo for the output
- /// @return Reference to the output TensorInfo.
+ /// @brief - Gets the matching TensorInfo for the output.
+ /// @return - References to the output TensorInfo.
const TensorInfo& GetTensorInfo() const { return m_TensorInfo; }
- /// @brief Get the allocated tensor memory.
- /// @return Pointer to the tensor memory
+ /// @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
+ /// 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 Allocate memory for all the tensors assigned to the handlers
- void AllocateTensors();
-
/// @brief Returns true if SetTensorInfo() has been called at least once on this.
bool IsTensorInfoSet() const { return m_bTensorInfoSet; }
private: