aboutsummaryrefslogtreecommitdiff
path: root/arm_compute/core/ITensorInfo.h
diff options
context:
space:
mode:
authorGeorgios Pinitas <georgios.pinitas@arm.com>2019-09-02 13:18:55 +0100
committerGeorgios Pinitas <georgios.pinitas@arm.com>2019-09-02 15:38:01 +0000
commit49be2e32e697f3b7a124018bc3cee91adb5f9478 (patch)
treebb6569c7142226aac093f15904d28073f579666d /arm_compute/core/ITensorInfo.h
parentae99b6eac40c7c3cb5eb465f3cbe4b522eff0488 (diff)
downloadComputeLibrary-49be2e32e697f3b7a124018bc3cee91adb5f9478.tar.gz
COMPMID-2644: Add is_dynamic flag to ITensorInfo.
Flag indicates if tensor shape is determined dynamically in kernel/function execution. Change-Id: I7f6f9557571316e1eaf8c7aaf9f5b9262c2e1751 Signed-off-by: Georgios Pinitas <georgios.pinitas@arm.com> Reviewed-on: https://review.mlplatform.org/c/1850 Tested-by: Arm Jenkins <bsgcomp@arm.com> Reviewed-by: Michalis Spyrou <michalis.spyrou@arm.com> Comments-Addressed: Arm Jenkins <bsgcomp@arm.com>
Diffstat (limited to 'arm_compute/core/ITensorInfo.h')
-rw-r--r--arm_compute/core/ITensorInfo.h16
1 files changed, 14 insertions, 2 deletions
diff --git a/arm_compute/core/ITensorInfo.h b/arm_compute/core/ITensorInfo.h
index f113445fb7..5aa243ecf8 100644
--- a/arm_compute/core/ITensorInfo.h
+++ b/arm_compute/core/ITensorInfo.h
@@ -1,5 +1,5 @@
/*
- * Copyright (c) 2016-2018 ARM Limited.
+ * Copyright (c) 2016-2019 ARM Limited.
*
* SPDX-License-Identifier: MIT
*
@@ -200,6 +200,11 @@ public:
* @return True if the tensor size can be changed.
*/
virtual bool is_resizable() const = 0;
+ /** Flag indicating whether the shape of the tensor is dynamic, meaning that it can change on kernel/function execution.
+ *
+ * @return True if its dynamic else false
+ */
+ virtual bool is_dynamic() const = 0;
/** Set the flag whether the tensor size can be changed.
*
* @param[in] is_resizable Flag that marks the tensor if it can be changed or not.
@@ -207,6 +212,13 @@ public:
* @return Reference to this ITensorInfo object
*/
virtual ITensorInfo &set_is_resizable(bool is_resizable) = 0;
+ /** Set the flag whether the tensor size is dynamic.
+ *
+ * @param[in] is_dynamic Flag that marks the tensor if it's dynamic.
+ *
+ * @return Reference to this ITensorInfo object
+ */
+ virtual ITensorInfo &set_is_dynamic(bool is_dynamic) = 0;
/** Valid region of the tensor. All elements in the valid region have defined values, i.e. are not undefined.
*
* @return The valid region.
@@ -274,5 +286,5 @@ public:
return std::pair<TensorShape, ValidRegion>(bc_shape, bc_valid_region);
}
};
-}
+} // namespace arm_compute
#endif /*__ARM_COMPUTE_TENSORINFO_H__ */