We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 12d0a04 commit 7390d5bCopy full SHA for 7390d5b
test/nn/nnlib.jl
@@ -78,18 +78,15 @@ end
78
end
79
80
@testset "conv 1d: flip" begin
81
- x = [1; 2; 3;;;]
82
- W = [1; 2; 3;;;]
83
-
+ x = [1.0f0; 2.0f0; 3.0f0;;;]
+ W = [1.0f0; 2.0f0; 3.0f0;;;]
84
xx = Reactant.ConcreteRArray(x)
85
WW = Reactant.ConcreteRArray(W)
86
87
conv_noflip(x, W) = NNlib.conv(x, W; pad=1, flipped=true)
88
conv_flip(x, W) = NNlib.conv(x, W; pad=1, flipped=false)
89
90
@test Reactant.compile(conv_noflip, (xx, WW))(xx, WW) ==
91
- [0*1+1*2+2*3; 1*1+2*2+3*3; 1*2+2*3+3*0;;;]
+ [0*1+1*2+2*3; 1*1+2*2+3*3; 1*2+2*3+3*0;;;]
92
@test Reactant.compile(conv_flip, (xx, WW))(xx, WW) ==
93
- [3*0+2*1+1*2; 3*1+2*2+1*3; 3*2+2*3+1*0;;;]
+ [3*0+2*1+1*2; 3*1+2*2+1*3; 3*2+2*3+1*0;;;]
94
95
0 commit comments