diff options
| author | Cedric Nugteren <web@cedricnugteren.nl> | 2017-08-23 19:54:09 +0200 |
|---|---|---|
| committer | Cedric Nugteren <web@cedricnugteren.nl> | 2017-08-23 19:54:09 +0200 |
| commit | a8c26594d9ee5b735b9d1b0400ac7eba4abf02fb (patch) | |
| tree | b58afbc60b9c0381ab054d5321df51ec92cfd5fa /test/routines/levelx | |
| parent | 803ca781f9be56f86a0806689f8886a2428d5b9f (diff) | |
Made the im2col client properly handle the arguments
Diffstat (limited to 'test/routines/levelx')
| -rw-r--r-- | test/routines/levelx/xim2col.hpp | 4 |
1 files changed, 3 insertions, 1 deletions
diff --git a/test/routines/levelx/xim2col.hpp b/test/routines/levelx/xim2col.hpp index e6aefd9e..59be8156 100644 --- a/test/routines/levelx/xim2col.hpp +++ b/test/routines/levelx/xim2col.hpp @@ -134,7 +134,9 @@ public: return 1; } static size_t GetBytes(const Arguments<T> &args) { - return (1) * sizeof(T); + const auto input = args.channels * args.width * args.height; // possibly less with striding + const auto output = args.kernel_h * args.kernel_w * NumPatches(args); + return (input + output) * sizeof(T); } }; |