diff options
Diffstat (limited to 'tests/validation/reference/Im2Col.h')
-rw-r--r-- | tests/validation/reference/Im2Col.h | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/tests/validation/reference/Im2Col.h b/tests/validation/reference/Im2Col.h index 5277171a2f..b1ebaf25da 100644 --- a/tests/validation/reference/Im2Col.h +++ b/tests/validation/reference/Im2Col.h @@ -35,7 +35,7 @@ namespace validation namespace reference { template <typename T> -void im2col(const SimpleTensor<T> &src, SimpleTensor<T> &dst, 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, bool channels_first_output_nhwc = false); } // namespace reference } // namespace validation } // namespace test |