diff options
Diffstat (limited to 'arm_compute/core/SubTensorInfo.h')
| -rw-r--r-- | arm_compute/core/SubTensorInfo.h | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/arm_compute/core/SubTensorInfo.h b/arm_compute/core/SubTensorInfo.h index b8a36854dc..7c464c0b17 100644 --- a/arm_compute/core/SubTensorInfo.h +++ b/arm_compute/core/SubTensorInfo.h @@ -185,10 +185,11 @@ public: ARM_COMPUTE_ERROR_ON(_parent == nullptr); return _parent->is_resizable(); } - void set_is_resizable(bool is_resizable) override + ITensorInfo &set_is_resizable(bool is_resizable) override { ARM_COMPUTE_ERROR_ON(_parent == nullptr); _parent->set_is_resizable(is_resizable); + return *this; } ValidRegion valid_region() const override { |