From d17a677a70d477ae508c084148ba6974fc38477d Mon Sep 17 00:00:00 2001 From: Isabella Gottardi Date: Tue, 27 Feb 2018 17:41:55 +0000 Subject: 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 Reviewed-by: Anthony Barbier Reviewed-by: Pablo Tello --- arm_compute/core/ITensorInfo.h | 12 ++++++++++++ 1 file changed, 12 insertions(+) (limited to 'arm_compute/core/ITensorInfo.h') 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. -- cgit v1.2.1