diff options
Diffstat (limited to 'src/armnn/backends/OutputHandler.hpp')
-rw-r--r-- | src/armnn/backends/OutputHandler.hpp | 21 |
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: |