Skip to content

Commit

Permalink
divceil vs div
Browse files Browse the repository at this point in the history
  • Loading branch information
mathieupoumeyrolsonos authored and kali committed Dec 4, 2023
1 parent 6e5c0f4 commit e6ad6d0
Showing 1 changed file with 2 additions and 1 deletion.
3 changes: 2 additions & 1 deletion core/src/ops/cnn/padding.rs
Original file line number Diff line number Diff line change
Expand Up @@ -192,10 +192,11 @@ impl PaddingSpec {
} else {
let kernel_field = (kernel - 1) * dilation + 1;
let dividend = input.clone() + bef + aft - kernel_field;
let output = dividend.divceil(stride) + 1;
let output = dividend.div(stride) + 1;
ComputedPaddedDim::new(input.clone(), output, bef.into(), aft.into())
}
}

fn explicit_usize(
input: usize,
kernel: usize,
Expand Down

0 comments on commit e6ad6d0

Please sign in to comment.