COMPMID-979: Add NHWC data layout to the tensor's metadata

Change-Id: I89bdab7dc63a437eb7d60c0ae652c53c3875e503
Reviewed-on: https://eu-gerrit-1.euhpc.arm.com/122524
Tested-by: Jenkins <bsgcomp@arm.com>
Reviewed-by: Anthony Barbier <anthony.barbier@arm.com>
Reviewed-by: Pablo Tello <pablo.tello@arm.com>

1 files changed, 12 insertions, 0 deletions

diff --git a/arm_compute/core/ITensorInfo.h b/arm_compute/core/ITensorInfo.h
index b5677dffd6..50a1eb2ff1 100644
--- a/arm_compute/core/ITensorInfo.h
+++ b/arm_compute/core/ITensorInfo.h
@@ -97,6 +97,13 @@ public:
      * @return Reference to this ITensorInfo object
      */
     virtual ITensorInfo &set_quantization_info(const QuantizationInfo &quantization_info) = 0;
+    /** Set the data layout of the tensor.
+     *
+     * @param[in] data_layout DataLayout containing the layout data information.
+     *
+     * @return Reference to this ITensorInfo object
+     */
+    virtual ITensorInfo &set_data_layout(const DataLayout &data_layout) = 0;
     /** Resets the padding settings of the tensor.
     *
     * @return Reference to this ITensorInfo object
@@ -222,6 +229,11 @@ public:
     * @return A QuantizationInfo containing the scale and offset.
     */
     virtual QuantizationInfo quantization_info() const = 0;
+    /** Get the data layout of the tensor.
+    *
+    * @return A DataLayout containing the layout data information.
+    */
+    virtual DataLayout data_layout() const = 0;
 
     /** If infos are broadcast compatible tensor info's, return the broadcasted shape and the intersection of
      * the broadcasted valid regions of the tensors.