diff options
Diffstat (limited to 'arm_compute/core/ITensor.h')
-rw-r--r-- | arm_compute/core/ITensor.h | 8 |
1 files changed, 5 insertions, 3 deletions
diff --git a/arm_compute/core/ITensor.h b/arm_compute/core/ITensor.h index abcec60b79..aad8313261 100644 --- a/arm_compute/core/ITensor.h +++ b/arm_compute/core/ITensor.h @@ -32,7 +32,7 @@ namespace arm_compute { class Coordinates; -/** Interface for Neon tensor */ +/** Interface for CPU tensor */ class ITensor { public: @@ -90,11 +90,13 @@ public: bool is_used() const; /** Marks a tensor as unused */ void mark_as_unused() const; + /** Marks a tensor as used */ + void mark_as_used() const; private: - mutable bool _is_used = { true }; /**< Flag that marks if the tensor is used or not */ + mutable bool _is_used = {true}; /**< Flag that marks if the tensor is used or not */ }; using IImage = ITensor; -} +} // namespace arm_compute #endif /*ARM_COMPUTE_ITENSOR_H */ |