diff options
Diffstat (limited to 'arm_compute/graph/backends/NEON/NESubTensorHandle.h')
| -rw-r--r-- | arm_compute/graph/backends/NEON/NESubTensorHandle.h | 21 |
1 files changed, 15 insertions, 6 deletions
diff --git a/arm_compute/graph/backends/NEON/NESubTensorHandle.h b/arm_compute/graph/backends/NEON/NESubTensorHandle.h index 11dcec60f3..101d3e6644 100644 --- a/arm_compute/graph/backends/NEON/NESubTensorHandle.h +++ b/arm_compute/graph/backends/NEON/NESubTensorHandle.h @@ -52,18 +52,27 @@ public: NESubTensorHandle(NESubTensorHandle &&) = default; /** Allow instances of this class to be moved */ NESubTensorHandle &operator=(NESubTensorHandle &&) = default; + /** Prevent instances of this class from being copied (As this class contains pointers) */ + NESubTensorHandle(const NESubTensorHandle &) = delete; + /** Prevent instances of this class from being copied (As this class contains pointers) */ + NESubTensorHandle &operator=(const NESubTensorHandle &) = delete; // Inherited overridden methods - void allocate() override; + void allocate() override; + void free() override; + void manage(IMemoryGroup *mg) override; + void map(bool blocking) override; + void unmap() override; + void release_if_unused() override; arm_compute::ITensor &tensor() override; const arm_compute::ITensor &tensor() const override; - void map(bool blocking) override; - void unmap() override; - void release_if_unused() override; - bool is_subtensor() const override; + ITensorHandle *parent_handle() override; + bool is_subtensor() const override; + Target target() const override; private: - arm_compute::SubTensor _sub_tensor; /**< Backend Sub-Tensor */ + arm_compute::SubTensor _sub_tensor; /**< Backend Sub-Tensor */ + ITensorHandle *_parent_handle; /**< Parent handle */ }; } // namespace backends } // namespace graph |