diff options
Diffstat (limited to 'arm_compute/core/SubTensorInfo.h')
| -rw-r--r-- | arm_compute/core/SubTensorInfo.h | 6 |
1 files changed, 6 insertions, 0 deletions
diff --git a/arm_compute/core/SubTensorInfo.h b/arm_compute/core/SubTensorInfo.h index 5fec11a2e8..b8a36854dc 100644 --- a/arm_compute/core/SubTensorInfo.h +++ b/arm_compute/core/SubTensorInfo.h @@ -104,6 +104,12 @@ public: _parent->set_quantization_info(quantization_info); return *this; } + ITensorInfo &reset_padding() override + { + ARM_COMPUTE_ERROR_ON(_parent == nullptr); + _parent->reset_padding(); + return *this; + } bool auto_padding() override { ARM_COMPUTE_ERROR_ON(_parent == nullptr); |