aboutsummaryrefslogtreecommitdiff
path: root/tests/validation/reference/Im2Col.h
diff options
context:
space:
mode:
Diffstat (limited to 'tests/validation/reference/Im2Col.h')
-rw-r--r--tests/validation/reference/Im2Col.h2
1 files changed, 1 insertions, 1 deletions
diff --git a/tests/validation/reference/Im2Col.h b/tests/validation/reference/Im2Col.h
index 4fe6ea9acf..5277171a2f 100644
--- a/tests/validation/reference/Im2Col.h
+++ b/tests/validation/reference/Im2Col.h
@@ -35,7 +35,7 @@ namespace validation
namespace reference
{
template <typename T>
-SimpleTensor<T> im2col(const SimpleTensor<T> &src, const TensorShape &dst_shape, const Size2D &kernel_dims, const PadStrideInfo &conv_info, bool has_bias);
+void im2col(const SimpleTensor<T> &src, SimpleTensor<T> &dst, const Size2D &kernel_dims, const PadStrideInfo &conv_info, bool has_bias);
} // namespace reference
} // namespace validation
} // namespace test