Restrict types of arguments

Co-authored-by: David Widmann <[email protected]>
JuliaDiff · Sep 26, 2023 · 7d2ef0a · 7d2ef0a
1 parent a3f0527
commit 7d2ef0a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/rulesets/LinearAlgebra/dense.jl b/src/rulesets/LinearAlgebra/dense.jl
@@ -399,7 +399,7 @@ end
 ##### `kron`
 #####
 
-function frule((_, Δx, Δy), ::typeof(kron), x, y)
+function frule((_, Δx, Δy), ::typeof(kron), x::AbstractVecOrMat{<:Number}, y::AbstractVecOrMat{<:Number})
     return kron(x, y), kron(Δx, y) + kron(x, Δy)
 end