diff options
| author | Michalis Spyrou <michalis.spyrou@arm.com> | 2018-03-02 10:25:59 +0000 |
|---|---|---|
| committer | Anthony Barbier <anthony.barbier@arm.com> | 2018-11-02 16:48:33 +0000 |
| commit | cf581f56cf7e33d6d856b3f361af2f050ba28a6c (patch) | |
| tree | a66d9d1e136a3cc7c9c6d39cf142b86b3a3e6f2e /tests/Utils.h | |
| parent | 577fbdf87136626df68f67a1d61aca8daa5bfbfa (diff) | |
| download | ComputeLibrary-cf581f56cf7e33d6d856b3f361af2f050ba28a6c.tar.gz | |
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 <michele.digiorgio@arm.com>
Tested-by: Jenkins <bsgcomp@arm.com>
Diffstat (limited to 'tests/Utils.h')
| -rw-r--r-- | tests/Utils.h | 4 |
1 files changed, 3 insertions, 1 deletions
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 <typename T> 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; |