commit d56e770e7c394d13706a21ee350e7dafe4278987
author Isabella Gottardi <isabella.gottardi@arm.com> Wed Feb 28 14:29:36 2018 +0000
committer Anthony Barbier <anthony.barbier@arm.com> Fri Nov 02 16:48:33 2018 +0000
tree 2f06e9b2857a9020ea2c1d60982112bb2eabda4c
parent 847864df46cea578bb0766c17c95ad5e804c49c3

COMPMID-979: Add NHWC data layout to the tensor's metadata (Part 2)

Change-Id: I24aa35a85834abf0c9954aba714aeae654615b44
Reviewed-on: https://eu-gerrit-1.euhpc.arm.com/122646
Tested-by: Jenkins <bsgcomp@arm.com>
Reviewed-by: Anthony Barbier <anthony.barbier@arm.com>

arm_compute/core/SubTensorInfo.h

diff --git a/arm_compute/core/SubTensorInfo.h b/arm_compute/core/SubTensorInfo.h
index f9ed99b..882e4ec 100644
--- a/arm_compute/core/SubTensorInfo.h
+++ b/arm_compute/core/SubTensorInfo.h
@@ -127,6 +127,11 @@
     {
         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);