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