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>
diff --git a/arm_compute/core/TensorInfo.h b/arm_compute/core/TensorInfo.h
index 1eaf052..77913f5 100644
--- a/arm_compute/core/TensorInfo.h
+++ b/arm_compute/core/TensorInfo.h
@@ -1,5 +1,5 @@
/*
- * Copyright (c) 2016-2018 ARM Limited.
+ * Copyright (c) 2016-2019 ARM Limited.
*
* SPDX-License-Identifier: MIT
*
@@ -277,11 +277,20 @@
{
return _is_resizable;
}
+ bool is_dynamic() const override
+ {
+ return _is_dynamic;
+ }
ITensorInfo &set_is_resizable(bool is_resizable) override
{
_is_resizable = is_resizable;
return *this;
}
+ ITensorInfo &set_is_dynamic(bool is_dynamic) override
+ {
+ _is_dynamic = is_dynamic;
+ return *this;
+ }
ValidRegion valid_region() const override
{
return _valid_region;
@@ -314,10 +323,11 @@
DataType _data_type;
Format _format;
bool _is_resizable;
+ bool _is_dynamic;
ValidRegion _valid_region;
PaddingSize _padding;
QuantizationInfo _quantization_info;
DataLayout _data_layout;
};
-}
+} // namespace arm_compute
#endif /*__ARM_COMPUTE_TENSORINFO_H__ */