|
647 | 647 | @test size(conv(x, w)) == (9, 9, 16, 10)
|
648 | 648 | @test size(conv(x, w; stride = (2, 2), pad = (2, 2))) == (7, 7, 16, 10)
|
649 | 649 | @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) |
650 | 652 | end
|
651 | 653 |
|
652 | 654 | @testset "depthwiseconv_wrapper" begin
|
|
656 | 658 | @test size(depthwiseconv(x, w)) == (9, 9, 9, 10)
|
657 | 659 | @test size(depthwiseconv(x, w; stride = (2, 2), pad = (2, 2))) == (7, 7, 9, 10)
|
658 | 660 | @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) |
659 | 663 | end
|
0 commit comments