diff options
Diffstat (limited to 'arm_compute/runtime/Tensor.h')
-rw-r--r-- | arm_compute/runtime/Tensor.h | 10 |
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 */ |