diff options
Diffstat (limited to 'tests/Utils.h')
| -rw-r--r-- | tests/Utils.h | 18 |
1 files changed, 17 insertions, 1 deletions
diff --git a/tests/Utils.h b/tests/Utils.h index 750d907778..27e0397b6b 100644 --- a/tests/Utils.h +++ b/tests/Utils.h @@ -1,5 +1,5 @@ /* - * Copyright (c) 2017, 2018 ARM Limited. + * Copyright (c) 2017-2018 ARM Limited. * * SPDX-License-Identifier: MIT * @@ -500,6 +500,22 @@ inline T create_tensor(const TensorShape &shape, Format format) return tensor; } +/** Create and initialize a multi-image of the given type. + * + * @param[in] shape Tensor shape. + * @param[in] format Format type. + * + * @return Initialized tensor of given type. + */ +template <typename T> +inline T create_multi_image(const TensorShape &shape, Format format) +{ + T multi_image; + multi_image.init(shape.x(), shape.y(), format); + + return multi_image; +} + /** Create and initialize a HOG (Histogram of Oriented Gradients) of the given type. * * @param[in] cell_size Cell size in pixels |