From 97bcf77d4bc9b31e32a8785787e0497ac5440e44 Mon Sep 17 00:00:00 2001
From: Cedric Nugteren <web@cedricnugteren.nl>
Date: Sun, 16 Jul 2017 22:33:49 +0200
Subject: First step towards supporting im2col in the test infrastructure

---
 src/utilities/utilities.hpp | 25 +++++++++++++++++++++++++
 1 file changed, 25 insertions(+)

(limited to 'src/utilities/utilities.hpp')

diff --git a/src/utilities/utilities.hpp b/src/utilities/utilities.hpp
index 03051354..8daeda08 100644
--- a/src/utilities/utilities.hpp
+++ b/src/utilities/utilities.hpp
@@ -78,6 +78,19 @@ constexpr auto kArgAlpha = "alpha";
 constexpr auto kArgBeta = "beta";
 constexpr auto kArgBatchCount = "batch_num";
 
+// Constants for im2col
+constexpr auto kArgChannels = "channels";
+constexpr auto kArgHeight = "height";
+constexpr auto kArgWidth = "width";
+constexpr auto kArgKernelH = "kernelh";
+constexpr auto kArgKernelW = "kernelw";
+constexpr auto kArgPadH = "padh";
+constexpr auto kArgPadW = "padw";
+constexpr auto kArgStrideH = "strideh";
+constexpr auto kArgStrideW = "stridew";
+constexpr auto kArgDilationH = "dilationh";
+constexpr auto kArgDilationW = "dilationw";
+
 // The tuner-specific arguments in string form
 constexpr auto kArgFraction = "fraction";
 
@@ -170,6 +183,18 @@ struct Arguments {
   size_t imax_offset = 0;
   T alpha = ConstantOne<T>();
   T beta = ConstantOne<T>();
+  // Arguments for im2col
+  size_t channels = 1;
+  size_t height = 1;
+  size_t width = 1;
+  size_t kernel_h = 1;
+  size_t kernel_w = 1;
+  size_t pad_h = 1;
+  size_t pad_w = 1;
+  size_t stride_h = 1;
+  size_t stride_w = 1;
+  size_t dilation_h = 1;
+  size_t dilation_w = 1;
   // Batch-specific arguments
   size_t batch_count = 1;
   std::vector<size_t> x_offsets; // = {0};
-- 
cgit v1.2.3