1 files changed, 7 insertions, 3 deletions
diff --git a/arm_compute/runtime/Tensor.h b/arm_compute/runtime/Tensor.h
index 6fa7c8ca0e..e469f70817 100644
--- a/arm_compute/runtime/Tensor.h
+++ b/arm_compute/runtime/Tensor.h
@@ -32,13 +32,17 @@
 namespace arm_compute
 {
 class ITensorInfo;
-
+class IRuntimeContext;
 /** Basic implementation of the tensor interface */
 class Tensor : public ITensor, public IMemoryManageable
 {
 public:
-    /** Constructor */
-    Tensor();
+    /** Constructor
+     *
+     * @param[in] ctx (Optional) Pointer to the runtime context.
+     *
+     */
+    Tensor(IRuntimeContext *ctx = nullptr);
     /** Destructor: free the tensor's memory */
     ~Tensor() = default;
     /** Allow instances of this class to be move constructed */