Skip to content

Commit 3ac9933

Browse files
committed
Added tests with dilation and flipkernel
1 parent d74af94 commit 3ac9933

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

test/conv.jl

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -647,6 +647,8 @@ end
647647
@test size(conv(x, w)) == (9, 9, 16, 10)
648648
@test size(conv(x, w; stride = (2, 2), pad = (2, 2))) == (7, 7, 16, 10)
649649
@test size(conv(x, w1; stride = (1, 2), pad = (2, 3))) == (12, 7, 16, 10)
650+
@test size(conv(x, w; stride = (1, 2), pad = (2, 3), dilation = (2, 2))) == (12, 7, 16, 10)
651+
@test size(conv(x, w; stride = (1, 2), pad = (2, 3), dilation = (2, 2), flipped = true)) == (12, 7, 16, 10)
650652
end
651653

652654
@testset "depthwiseconv_wrapper" begin
@@ -656,4 +658,6 @@ end
656658
@test size(depthwiseconv(x, w)) == (9, 9, 9, 10)
657659
@test size(depthwiseconv(x, w; stride = (2, 2), pad = (2, 2))) == (7, 7, 9, 10)
658660
@test size(depthwiseconv(x, w1; stride = (1, 2), pad = (2, 3))) == (12, 7, 9, 10)
661+
@test size(depthwiseconv(x, w1; stride = (1, 2), pad = (2, 3), dilation = (2, 2))) == (10, 5, 9, 10)
662+
@test size(depthwiseconv(x, w1; stride = (1, 2), pad = (2, 3), dilation = (2, 2), flipped = true)) == (10, 5, 9, 10)
659663
end

0 commit comments

Comments
 (0)