diff options
Diffstat (limited to 'arm_compute/core/SubTensorInfo.h')
| -rw-r--r-- | arm_compute/core/SubTensorInfo.h | 5 |
1 files changed, 5 insertions, 0 deletions
diff --git a/arm_compute/core/SubTensorInfo.h b/arm_compute/core/SubTensorInfo.h index f9ed99b308..882e4ec1d0 100644 --- a/arm_compute/core/SubTensorInfo.h +++ b/arm_compute/core/SubTensorInfo.h @@ -127,6 +127,11 @@ public: { return _tensor_shape[index]; } + size_t dimension(DataLayoutDimension dimension) const override + { + ARM_COMPUTE_ERROR_ON(_parent == nullptr); + return get_data_layout_dimension_index(_parent->data_layout(), dimension); + } const Strides &strides_in_bytes() const override { ARM_COMPUTE_ERROR_ON(_parent == nullptr); |