diff options
Diffstat (limited to 'src/core/NEON/kernels/arm_conv/depthwise/depthwise_depthfirst_generic.hpp')
| -rw-r--r-- | src/core/NEON/kernels/arm_conv/depthwise/depthwise_depthfirst_generic.hpp | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/src/core/NEON/kernels/arm_conv/depthwise/depthwise_depthfirst_generic.hpp b/src/core/NEON/kernels/arm_conv/depthwise/depthwise_depthfirst_generic.hpp index ca5026b6e0..e2d05560a1 100644 --- a/src/core/NEON/kernels/arm_conv/depthwise/depthwise_depthfirst_generic.hpp +++ b/src/core/NEON/kernels/arm_conv/depthwise/depthwise_depthfirst_generic.hpp @@ -186,7 +186,7 @@ class GenericInputArrayElement static size_t get_element_size(const WorkspaceArgs<IDepthfirstStrategy, OutputStage> &args) { const auto kernel_points = args.depthwise_args.kernel_rows * args.depthwise_args.kernel_cols; - return sizeof(T **) * args.strategy->get_input_rows() * args.strategy->get_input_cols() * kernel_points; + return sizeof(T **) * args.strategy->get_output_rows() * args.strategy->get_output_cols() * kernel_points; } template <class WorkspaceType, class OutputStage> |