From 1562be3e8a449360a90af75f6f1481a30d41be75 Mon Sep 17 00:00:00 2001 From: Georgios Pinitas Date: Thu, 8 Mar 2018 19:09:19 +0000 Subject: COMPMID-998: Release unused trainable parameters. Change-Id: I361a520f34080016a25bc86e1e6789777c5152c1 Reviewed-on: https://eu-gerrit-1.euhpc.arm.com/124432 Reviewed-by: Anthony Barbier Tested-by: Jenkins --- arm_compute/core/ITensor.h | 10 ++++++++++ 1 file changed, 10 insertions(+) (limited to 'arm_compute/core') diff --git a/arm_compute/core/ITensor.h b/arm_compute/core/ITensor.h index 1ef9c6d3f6..255a236861 100644 --- a/arm_compute/core/ITensor.h +++ b/arm_compute/core/ITensor.h @@ -83,6 +83,16 @@ public: * @param io_fmt Format information */ void print(std::ostream &s, IOFormatInfo io_fmt = IOFormatInfo()) const; + /** Flags if the tensor is used or not + * + * @return True if it is used else false + */ + bool is_used() const; + /** Marks a tensor as unused */ + void mark_as_unused() const; + +private: + mutable bool _is_used = { true }; /**< Flag that marks if the tensor is used or not */ }; using IImage = ITensor; -- cgit v1.2.1