aboutsummaryrefslogtreecommitdiff
path: root/arm_compute/runtime/CL/CLTensor.h
diff options
context:
space:
mode:
Diffstat (limited to 'arm_compute/runtime/CL/CLTensor.h')
-rw-r--r--arm_compute/runtime/CL/CLTensor.h5
1 files changed, 3 insertions, 2 deletions
diff --git a/arm_compute/runtime/CL/CLTensor.h b/arm_compute/runtime/CL/CLTensor.h
index 65ff4f2bba..bc72839492 100644
--- a/arm_compute/runtime/CL/CLTensor.h
+++ b/arm_compute/runtime/CL/CLTensor.h
@@ -37,7 +37,7 @@ class ITensorAllocator;
class ITensorInfo;
/** Basic implementation of the OpenCL tensor interface */
-class CLTensor : public ICLTensor
+class CLTensor : public ICLTensor, public IMemoryManageable
{
public:
/** Constructor */
@@ -68,6 +68,7 @@ public:
TensorInfo *info() override;
const cl::Buffer &cl_buffer() const override;
CLQuantization quantization() const override;
+ void associate_memory_group(IMemoryGroup *memory_group) override;
protected:
// Inherited methods overridden:
@@ -80,5 +81,5 @@ private:
/** OpenCL Image */
using CLImage = CLTensor;
-}
+} // namespace arm_compute
#endif /*__ARM_COMPUTE_CLTENSOR_H__ */