COMPMID-998: Release unused trainable parameters.

Change-Id: I361a520f34080016a25bc86e1e6789777c5152c1
Reviewed-on: https://eu-gerrit-1.euhpc.arm.com/124432
Reviewed-by: Anthony Barbier <anthony.barbier@arm.com>
Tested-by: Jenkins <bsgcomp@arm.com>

1 files changed, 10 insertions, 0 deletions

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;