diff options
Diffstat (limited to 'tests/validation/reference/Winograd.h')
| -rw-r--r-- | tests/validation/reference/Winograd.h | 3 |
1 files changed, 3 insertions, 0 deletions
diff --git a/tests/validation/reference/Winograd.h b/tests/validation/reference/Winograd.h index ba8e5c1cb6..fa1a7f3f61 100644 --- a/tests/validation/reference/Winograd.h +++ b/tests/validation/reference/Winograd.h @@ -41,6 +41,9 @@ SimpleTensor<T> winograd_input_transform(const SimpleTensor<T> &src, const Tenso template <typename T> SimpleTensor<T> winograd_filter_transform(const SimpleTensor<T> &in, const TensorShape &output_shape); + +template <typename T> +SimpleTensor<T> winograd_output_transform(const SimpleTensor<T> &in, const TensorShape &output_shape, const Size2D &kernel_dims, const Size2D &num_tiles); } // namespace reference } // namespace validation } // namespace test |