diff options
author | Cedric Nugteren <web@cedricnugteren.nl> | 2018-05-06 11:35:34 +0200 |
---|---|---|
committer | Cedric Nugteren <web@cedricnugteren.nl> | 2018-05-06 11:35:34 +0200 |
commit | 2d1f6ba7fe842ba938490fc599b6ebd209b6560b (patch) | |
tree | f1a284e5dc0163b7fed938a3efeb39432b9d3788 /src/utilities | |
parent | 2776d761768295b01a8be7c333dbb337805d7f77 (diff) |
Added convgemm skeleton, test infrastructure, and first reference implementation
Diffstat (limited to 'src/utilities')
-rw-r--r-- | src/utilities/utilities.hpp | 4 |
1 files changed, 3 insertions, 1 deletions
diff --git a/src/utilities/utilities.hpp b/src/utilities/utilities.hpp index 0edf77fe..2d2cd62e 100644 --- a/src/utilities/utilities.hpp +++ b/src/utilities/utilities.hpp @@ -84,6 +84,7 @@ constexpr auto kArgImaxOffset = "offimax"; constexpr auto kArgAlpha = "alpha"; constexpr auto kArgBeta = "beta"; constexpr auto kArgBatchCount = "batch_num"; +constexpr auto kArgNumKernels = "num_kernels"; // Constants for im2col constexpr auto kArgChannels = "channels"; @@ -195,7 +196,7 @@ struct Arguments { size_t imax_offset = 0; T alpha = ConstantOne<T>(); T beta = ConstantOne<T>(); - // Arguments for im2col + // Arguments for im2col and convgemm size_t channels = 1; size_t height = 1; size_t width = 1; @@ -207,6 +208,7 @@ struct Arguments { size_t stride_w = 1; size_t dilation_h = 1; size_t dilation_w = 1; + size_t num_kernels = 1; // Batch-specific arguments size_t batch_count = 1; std::vector<size_t> x_offsets; // = {0}; |