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/SubTensorInfo.h | 11 +++++++++++ 1 file changed, 11 insertions(+) (limited to 'arm_compute/core/SubTensorInfo.h') diff --git a/arm_compute/core/SubTensorInfo.h b/arm_compute/core/SubTensorInfo.h index 7f4239d49b..f9ed99b308 100644 --- a/arm_compute/core/SubTensorInfo.h +++ b/arm_compute/core/SubTensorInfo.h @@ -80,6 +80,12 @@ public: _parent->set_data_type(data_type); return *this; }; + ITensorInfo &set_data_layout(const DataLayout &data_layout) override + { + ARM_COMPUTE_ERROR_ON(_parent == nullptr); + _parent->set_data_layout(data_layout); + return *this; + }; ITensorInfo &set_num_channels(int num_channels) override { ARM_COMPUTE_ERROR_ON(_parent == nullptr); @@ -211,6 +217,11 @@ public: ARM_COMPUTE_ERROR_ON(_parent == nullptr); return _parent->quantization_info(); } + DataLayout data_layout() const override + { + ARM_COMPUTE_ERROR_ON(_parent == nullptr); + return _parent->data_layout(); + } private: ITensorInfo *_parent; -- cgit v1.2.1