use muladd

impICNF · Sep 13, 2022 · dde8838 · dde8838
1 parent 2eeceb0
commit dde8838
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/planar.jl b/src/planar.jl
@@ -25,7 +25,7 @@ Flux.@functor PlanarNN (u, w, b)
 function (m::PlanarNN)(z::AbstractVecOrMat)::AbstractVecOrMat
     u, w, b = m.u, m.w, only(m.b)
     h = NNlib.fast_act(m.h, z)
-    u * h.(transpose(w) * z .+ b)
+    u * h.(muladd(transpose(w), z, b))
 end
 
 """