From cf581f56cf7e33d6d856b3f361af2f050ba28a6c Mon Sep 17 00:00:00 2001 From: Michalis Spyrou Date: Fri, 2 Mar 2018 10:25:59 +0000 Subject: COMPMID-959 Add an optional DataLayout argument in create_tensor. Change-Id: Iecc90b48a3b64c7b812c1dea6f83604f76c50b83 Reviewed-on: https://eu-gerrit-1.euhpc.arm.com/122967 Reviewed-by: Michele DiGiorgio Tested-by: Jenkins --- tests/Utils.h | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) (limited to 'tests/Utils.h') diff --git a/tests/Utils.h b/tests/Utils.h index 27e0397b6b..5d24f5c0a1 100644 --- a/tests/Utils.h +++ b/tests/Utils.h @@ -467,16 +467,18 @@ inline bool is_in_valid_region(const ValidRegion &valid_region, Coordinates coor * @param[in] num_channels (Optional) Number of channels. * @param[in] fixed_point_position (Optional) Number of fractional bits. * @param[in] quantization_info (Optional) Quantization info for asymmetric quantized types. + * @param[in] data_layout (Optional) Data layout. Default is NCHW. * * @return Initialized tensor of given type. */ template inline T create_tensor(const TensorShape &shape, DataType data_type, int num_channels = 1, - int fixed_point_position = 0, QuantizationInfo quantization_info = QuantizationInfo()) + int fixed_point_position = 0, QuantizationInfo quantization_info = QuantizationInfo(), DataLayout data_layout = DataLayout::NCHW) { T tensor; TensorInfo info(shape, num_channels, data_type, fixed_point_position); info.set_quantization_info(quantization_info); + info.set_data_layout(data_layout); tensor.allocator()->init(info); return tensor; -- cgit v1.2.1