diff options
Diffstat (limited to 'tests/NEON')
-rw-r--r-- | tests/NEON/Helper.h | 15 |
1 files changed, 15 insertions, 0 deletions
diff --git a/tests/NEON/Helper.h b/tests/NEON/Helper.h index 8bd11cc57b..93f2010b95 100644 --- a/tests/NEON/Helper.h +++ b/tests/NEON/Helper.h @@ -64,6 +64,21 @@ public: } }; +// 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: + template <typename T, typename... Args> + void configure(T first, Args &&... args) + { + auto k = arm_compute::support::cpp14::make_unique<K>(); + k->configure(first, std::forward<Args>(args)...); + _kernel = std::move(k); + _border_handler.configure(first, BorderSize(bordersize), BorderMode::CONSTANT, PixelValue(0)); + } +}; + } // namespace test } // namespace arm_compute #endif /* __ARM_COMPUTE_TEST_NEON_HELPER_H__ */ |