aboutsummaryrefslogtreecommitdiff
path: root/arm_compute/core/ITensor.h
diff options
context:
space:
mode:
Diffstat (limited to 'arm_compute/core/ITensor.h')
-rw-r--r--arm_compute/core/ITensor.h10
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;