diff --git a/arm_compute/core/SubTensorInfo.h b/arm_compute/core/SubTensorInfo.h
index 7f4239d..f9ed99b 100644
--- a/arm_compute/core/SubTensorInfo.h
+++ b/arm_compute/core/SubTensorInfo.h
@@ -80,6 +80,12 @@
         _parent->set_data_type(data_type);
         return *this;
     };
+    ITensorInfo &set_data_layout(const DataLayout &data_layout) override
+    {
+        ARM_COMPUTE_ERROR_ON(_parent == nullptr);
+        _parent->set_data_layout(data_layout);
+        return *this;
+    };
     ITensorInfo &set_num_channels(int num_channels) override
     {
         ARM_COMPUTE_ERROR_ON(_parent == nullptr);
@@ -211,6 +217,11 @@
         ARM_COMPUTE_ERROR_ON(_parent == nullptr);
         return _parent->quantization_info();
     }
+    DataLayout data_layout() const override
+    {
+        ARM_COMPUTE_ERROR_ON(_parent == nullptr);
+        return _parent->data_layout();
+    }
 
 private:
     ITensorInfo *_parent;