diff options
Diffstat (limited to 'tests/NEON/Helper.h')
| -rw-r--r-- | tests/NEON/Helper.h | 15 |
1 files changed, 12 insertions, 3 deletions
diff --git a/tests/NEON/Helper.h b/tests/NEON/Helper.h index 93f2010b95..b9f961243a 100644 --- a/tests/NEON/Helper.h +++ b/tests/NEON/Helper.h @@ -1,5 +1,5 @@ /* - * Copyright (c) 2017 ARM Limited. + * Copyright (c) 2017-2018 ARM Limited. * * SPDX-License-Identifier: MIT * @@ -50,11 +50,15 @@ void fill_tensors(D &&dist, std::initializer_list<int> seeds, T &&tensor, Ts &&. } } -// This template synthetizes an INESimpleFunction which runs the given kernel K +/** This template synthetizes an INESimpleFunction which runs the given kernel K */ template <typename K> class NESynthetizeFunction : public INESimpleFunction { public: + /** Configure the kernel. + * + * @param[in] args Configuration arguments. + */ template <typename... Args> void configure(Args &&... args) { @@ -64,11 +68,16 @@ public: } }; -// As above but this also setups a Zero border on the input tensor of the specified bordersize +/** As above but this also setups a Zero border on the input tensor of the specified bordersize */ template <typename K, int bordersize> class NESynthetizeFunctionWithZeroConstantBorder : public INESimpleFunction { public: + /** Configure the kernel. + * + * @param[in] first First configuration argument. + * @param[in] args Rest of the configuration arguments. + */ template <typename T, typename... Args> void configure(T first, Args &&... args) { |