Skip to content
Snippets Groups Projects
Commit 37a2115e authored by Maxence Naud's avatar Maxence Naud Committed by Grégoire Kubler
Browse files

[fix] bad dimension computation in ConvDepthWise

parent 63065668
No related branches found
No related tags found
1 merge request!110fix : v0.3.2
......@@ -135,12 +135,12 @@ void ConvDepthWiseImpl2D_cpu_forward_kernel(const std::array<DimSize_t, 2>& stri
// output H size
const DimSize_t dilated_kernel_x = dilationDims[0]*(kernelDims[0] - 1) + 1;
const std::size_t oxSize =
static_cast<std::size_t>(std::floor(static_cast<float>(inputDims[2] - dilated_kernel_x + strideDims[0]) /
static_cast<std::size_t>(std::floor(static_cast<float>(inputDims[2] - dilated_kernel_x - 1 + strideDims[0]) /
static_cast<float>(strideDims[0])));
// output W size
const DimSize_t dilated_kernel_y = dilationDims[1]*(kernelDims[1] - 1) + 1;
const std::size_t oySize =
static_cast<std::size_t>(std::floor(static_cast<float>(inputDims[3] - dilated_kernel_y + strideDims[1]) /
static_cast<std::size_t>(std::floor(static_cast<float>(inputDims[3] - dilated_kernel_y - 1 + strideDims[1]) /
static_cast<float>(strideDims[1])));
// TODO: kernel computation
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment