diff options
Diffstat (limited to 'arm_compute/core/TensorInfo.h')
| -rw-r--r-- | arm_compute/core/TensorInfo.h | 5 |
1 files changed, 5 insertions, 0 deletions
diff --git a/arm_compute/core/TensorInfo.h b/arm_compute/core/TensorInfo.h index 27cf5bae82..97f9d03dc7 100644 --- a/arm_compute/core/TensorInfo.h +++ b/arm_compute/core/TensorInfo.h @@ -28,6 +28,7 @@ #include "ITensorInfo.h" #include "arm_compute/core/Coordinates.h" +#include "arm_compute/core/Helpers.h" #include "arm_compute/core/Strides.h" #include "arm_compute/core/TensorShape.h" #include "arm_compute/core/Types.h" @@ -228,6 +229,10 @@ public: { return _tensor_shape[index]; } + size_t dimension(DataLayoutDimension dimension) const override + { + return get_data_layout_dimension_index(_data_layout, dimension); + } const Strides &strides_in_bytes() const override { return _strides_in_bytes; |