From f955d515c45d19d8e244ca8d8dba915e1ceb20da Mon Sep 17 00:00:00 2001 From: Michele Di Giorgio Date: Wed, 27 Feb 2019 14:26:51 +0000 Subject: COMPMID-1318: Implementing Winograd 7x7 NHWC on OpenCL -- Part I Change-Id: I94c3c886718076c6eee09be37a074a4bb0e54809 Signed-off-by: giuros01 Reviewed-on: https://review.mlplatform.org/c/868 Tested-by: Arm Jenkins Reviewed-by: Gian Marco Iodice Comments-Addressed: Arm Jenkins --- src/core/CL/CLHelpers.cpp | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) (limited to 'src/core/CL/CLHelpers.cpp') diff --git a/src/core/CL/CLHelpers.cpp b/src/core/CL/CLHelpers.cpp index 801347e200..f4ceca8200 100644 --- a/src/core/CL/CLHelpers.cpp +++ b/src/core/CL/CLHelpers.cpp @@ -214,7 +214,10 @@ bool cl_winograd_convolution_layer_supported(const Size2D &output_tile, const Si WinogradConfiguration(std::pair(4, 4), std::pair(3, 3)), WinogradConfiguration(std::pair(4, 4), std::pair(5, 5)), WinogradConfiguration(std::pair(4, 1), std::pair(5, 1)), - WinogradConfiguration(std::pair(1, 4), std::pair(1, 5)) + WinogradConfiguration(std::pair(1, 4), std::pair(1, 5)), + WinogradConfiguration(std::pair(1, 2), std::pair(1, 7)), + WinogradConfiguration(std::pair(2, 1), std::pair(7, 1)), + WinogradConfiguration(std::pair(2, 2), std::pair(7, 7)), }; auto p = std::make_pair(std::pair(output_tile.width, output_tile.height), -- cgit v1.2.1