diff options
Diffstat (limited to 'test/routines')
-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); } }; |