diff options
Diffstat (limited to 'arm_compute/core/ITensor.h')
-rw-r--r-- | arm_compute/core/ITensor.h | 10 |
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; |