From 247f52cfe337f7b2542b900e3d8cf122e9d4f11c Mon Sep 17 00:00:00 2001 From: Gian Marco Iodice Date: Thu, 22 Mar 2018 11:24:56 +0000 Subject: COMPMID-1013 - Create WinogradInfo data structure COMPMID-1014 - Refactoring Winograd's dataset Change-Id: I6abdcbf9a90d663f4db666cd410afece9f1d034d Reviewed-on: https://eu-gerrit-1.euhpc.arm.com/125899 Tested-by: Jenkins Reviewed-by: Anthony Barbier --- tests/validation/reference/Winograd.h | 14 +++++++++++--- 1 file changed, 11 insertions(+), 3 deletions(-) (limited to 'tests/validation/reference/Winograd.h') diff --git a/tests/validation/reference/Winograd.h b/tests/validation/reference/Winograd.h index 62e136b09d..29181f1142 100644 --- a/tests/validation/reference/Winograd.h +++ b/tests/validation/reference/Winograd.h @@ -36,14 +36,22 @@ namespace validation { namespace reference { +/** Winograd transform type */ +enum class WinogradTransformType +{ + INPUT, /**< Winograd input transform */ + FILTER, /**< Winograd filter transform */ + OUTPUT /**< Winograd output transform */ +}; + template -SimpleTensor winograd_input_transform(const SimpleTensor &src, const TensorShape &dst_shape, const PadStrideInfo &conv_info, const Size2D &kernel_dims); +SimpleTensor winograd_input_transform(const SimpleTensor &in, const TensorShape &output_shape, const WinogradInfo &winograd_info); template -SimpleTensor winograd_filter_transform(const SimpleTensor &in, const TensorShape &output_shape, const Size2D &output_tile); +SimpleTensor winograd_filter_transform(const SimpleTensor &in, const TensorShape &output_shape, const WinogradInfo &winograd_info); template -SimpleTensor winograd_output_transform(const SimpleTensor &in, const TensorShape &output_shape, const Size2D &kernel_dims, const Size2D &num_tiles); +SimpleTensor winograd_output_transform(const SimpleTensor &in, const TensorShape &output_shape, const WinogradInfo &winograd_info); } // namespace reference } // namespace validation } // namespace test -- cgit v1.2.1