diff options
Diffstat (limited to 'tests/validation/reference/Winograd.h')
| -rw-r--r-- | tests/validation/reference/Winograd.h | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/tests/validation/reference/Winograd.h b/tests/validation/reference/Winograd.h index fa1a7f3f61..62e136b09d 100644 --- a/tests/validation/reference/Winograd.h +++ b/tests/validation/reference/Winograd.h @@ -40,7 +40,7 @@ template <typename T> SimpleTensor<T> winograd_input_transform(const SimpleTensor<T> &src, const TensorShape &dst_shape, const PadStrideInfo &conv_info, const Size2D &kernel_dims); template <typename T> -SimpleTensor<T> winograd_filter_transform(const SimpleTensor<T> &in, const TensorShape &output_shape); +SimpleTensor<T> winograd_filter_transform(const SimpleTensor<T> &in, const TensorShape &output_shape, const Size2D &output_tile); template <typename T> SimpleTensor<T> winograd_output_transform(const SimpleTensor<T> &in, const TensorShape &output_shape, const Size2D &kernel_dims, const Size2D &num_tiles); |