diff options
author | Isabella Gottardi <isabella.gottardi@arm.com> | 2018-02-27 17:41:55 +0000 |
---|---|---|
committer | Anthony Barbier <anthony.barbier@arm.com> | 2018-11-02 16:48:33 +0000 |
commit | d17a677a70d477ae508c084148ba6974fc38477d (patch) | |
tree | df6719551b1125941ecdfe4889826b0daa92088b /arm_compute/core/TensorInfo.h | |
parent | 240cfa65322da0c9b2ec0e78960ff4bd87c7573e (diff) | |
download | ComputeLibrary-d17a677a70d477ae508c084148ba6974fc38477d.tar.gz |
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>
Diffstat (limited to 'arm_compute/core/TensorInfo.h')
-rw-r--r-- | arm_compute/core/TensorInfo.h | 6 |
1 files changed, 6 insertions, 0 deletions
diff --git a/arm_compute/core/TensorInfo.h b/arm_compute/core/TensorInfo.h index 0b8989f942..27cf5bae82 100644 --- a/arm_compute/core/TensorInfo.h +++ b/arm_compute/core/TensorInfo.h @@ -220,6 +220,7 @@ public: ITensorInfo &set_tensor_shape(const TensorShape &shape) override; ITensorInfo &set_fixed_point_position(int fixed_point_position) override; ITensorInfo &set_quantization_info(const QuantizationInfo &quantization_info) override; + ITensorInfo &set_data_layout(const DataLayout &data_layout) override; ITensorInfo &reset_padding() override; bool auto_padding() override; bool extend_padding(const PaddingSize &padding) override; @@ -297,6 +298,10 @@ public: { return _quantization_info; } + DataLayout data_layout() const override + { + return _data_layout; + } private: /** Calculates strides, offset and total size resulting from the specified padding around the XY plane. @@ -317,6 +322,7 @@ private: ValidRegion _valid_region; PaddingSize _padding; QuantizationInfo _quantization_info; + DataLayout _data_layout; }; } #endif /*__ARM_COMPUTE_TENSORINFO_H__ */ |