You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
Hi!
Is there any particular reason that sigmoid_fast is not used in the helper function _gru_output in the GRU layer in reccurent.jl? I see it used in the LSTM layer.
function _gru_output(gxs, ghs, bs)
r = @. σ(gxs[1] + ghs[1] + bs[1])
z = @. σ(gxs[2] + ghs[2] + bs[2])
return r, z
end
In my (not so simple) use case i got a 5% speed up in the forward step using sigmoid_fast on cpu compared to the above version.
The text was updated successfully, but these errors were encountered:
Hi!
Is there any particular reason that
sigmoid_fast
is not used in the helper function_gru_output
in the GRU layer in reccurent.jl? I see it used in the LSTM layer.In my (not so simple) use case i got a 5% speed up in the forward step using
sigmoid_fast
on cpu compared to the above version.The text was updated successfully, but these errors were encountered: