aboutsummaryrefslogtreecommitdiff
path: root/arm_compute/runtime/Tensor.h
diff options
context:
space:
mode:
Diffstat (limited to 'arm_compute/runtime/Tensor.h')
-rw-r--r--arm_compute/runtime/Tensor.h7
1 files changed, 4 insertions, 3 deletions
diff --git a/arm_compute/runtime/Tensor.h b/arm_compute/runtime/Tensor.h
index 7022afff87..6fa7c8ca0e 100644
--- a/arm_compute/runtime/Tensor.h
+++ b/arm_compute/runtime/Tensor.h
@@ -1,5 +1,5 @@
/*
- * Copyright (c) 2016-2018 ARM Limited.
+ * Copyright (c) 2016-2019 ARM Limited.
*
* SPDX-License-Identifier: MIT
*
@@ -34,7 +34,7 @@ namespace arm_compute
class ITensorInfo;
/** Basic implementation of the tensor interface */
-class Tensor : public ITensor
+class Tensor : public ITensor, public IMemoryManageable
{
public:
/** Constructor */
@@ -55,6 +55,7 @@ public:
ITensorInfo *info() const override;
ITensorInfo *info() override;
uint8_t *buffer() const override;
+ void associate_memory_group(IMemoryGroup *memory_group) override;
private:
mutable TensorAllocator _allocator; /**< Instance of the basic CPU allocator.*/
@@ -62,5 +63,5 @@ private:
/** Image */
using Image = Tensor;
-}
+} // namespace arm_compute
#endif /*__ARM_COMPUTE_TENSOR_H__ */