Release 18.08

[platform/upstream/armnn.git] / src / armnn / backends / OutputHandler.hpp

diff --git a/src/armnn/backends/OutputHandler.hpp b/src/armnn/backends/OutputHandler.hpp
index 9cc87c6..ed95577 100644 (file)
--- 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: