From 97bcf77d4bc9b31e32a8785787e0497ac5440e44 Mon Sep 17 00:00:00 2001 From: Cedric Nugteren Date: Sun, 16 Jul 2017 22:33:49 +0200 Subject: First step towards supporting im2col in the test infrastructure --- test/correctness/tester.cpp | 5 +++++ 1 file changed, 5 insertions(+) (limited to 'test/correctness/tester.cpp') diff --git a/test/correctness/tester.cpp b/test/correctness/tester.cpp index fb85e609..1581fbfb 100644 --- a/test/correctness/tester.cpp +++ b/test/correctness/tester.cpp @@ -366,6 +366,11 @@ std::string Tester::GetOptionsString(const Arguments &args) { if (o == kArgAlpha) { result += kArgAlpha + equals + ToString(args.alpha) + " "; } if (o == kArgBeta) { result += kArgBeta + equals + ToString(args.beta) + " "; } if (o == kArgBatchCount){result += kArgBatchCount + equals + ToString(args.batch_count) + " "; } + if (o == kArgChannels) { result += kArgChannels + equals + ToString(args.channels) + " "; } + if (o == kArgHeight) { result += kArgHeight + equals + ToString(args.height) + " "; } + if (o == kArgWidth) { result += kArgWidth + equals + ToString(args.width) + " "; } + if (o == kArgKernelH) { result += kArgKernelH + equals + ToString(args.kernel_h) + " "; } + if (o == kArgKernelW) { result += kArgKernelW + equals + ToString(args.kernel_w) + " "; } } return result; } -- cgit v1.2.3