3 files changed, 11 insertions, 3 deletions
diff --git a/arm_compute/graph/INode.h b/arm_compute/graph/INode.h
index 1969423074..be4575de91 100644
--- a/arm_compute/graph/INode.h
+++ b/arm_compute/graph/INode.h
@@ -46,6 +46,8 @@ public:
      * @param[in] ctx    Graph context to be used
      * @param[in] input  Input tensor of the node
      * @param[in] output Output tensor of the node
+     *
+     * @return a pointer to the function which implements the node.
      */
     virtual std::unique_ptr<arm_compute::IFunction> instantiate_node(GraphContext &ctx, ITensorObject *input, ITensorObject *output) = 0;
     /** Override the existing target hint
diff --git a/arm_compute/graph/IOperation.h b/arm_compute/graph/IOperation.h
index a9fa4f83c7..b1afb21db7 100644
--- a/arm_compute/graph/IOperation.h
+++ b/arm_compute/graph/IOperation.h
@@ -1,5 +1,5 @@
 /*
- * Copyright (c) 2017 ARM Limited.
+ * Copyright (c) 2017-2018 ARM Limited.
  *
  * SPDX-License-Identifier: MIT
  *
@@ -43,6 +43,8 @@ public:
     /** Interface to be implemented that configures an operation
      *
      * @param[in] ctx Node parameters to be used by the operation
+     *
+     * @return a pointer to the function which implements the operation.
      */
     virtual std::unique_ptr<arm_compute::IFunction> configure(NodeContext &ctx) = 0;
     /** Interface to be implemented that returns the target of the operation
diff --git a/arm_compute/graph/ITensorObject.h b/arm_compute/graph/ITensorObject.h
index a922dd53fe..1b6f929305 100644
--- a/arm_compute/graph/ITensorObject.h
+++ b/arm_compute/graph/ITensorObject.h
@@ -1,5 +1,5 @@
 /*
- * Copyright (c) 2017 ARM Limited.
+ * Copyright (c) 2017-2018 ARM Limited.
  *
  * SPDX-License-Identifier: MIT
  *
@@ -61,7 +61,11 @@ public:
      *
      * @return Tensor
      */
-    virtual ITensor       *tensor()       = 0;
+    virtual ITensor *tensor() = 0;
+    /** Returns a pointer to the internal tensor
+     *
+     * @return const Tensor
+     */
     virtual const ITensor *tensor() const = 0;
     /** Return the target that this tensor is pinned on
      *