Update SymbolicsLuxCoreExt.jl

JuliaSymbolics · Sep 7, 2024 · a753920 · a753920
1 parent 754ba8a
commit a753920
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/ext/SymbolicsLuxCoreExt.jl b/ext/SymbolicsLuxCoreExt.jl
@@ -3,7 +3,7 @@ module SymbolicsLuxCoreExt
 using LuxCore, Symbolics
 
 @register_array_symbolic LuxCore.stateless_apply(
-    model::LuxCore.AbstractExplicitLayer, x::AbstractArray, ps::Union{NamedTuple, <:AbstractVector}) begin
+    model::LuxCore.AbstractLuxLayer, x::AbstractArray, ps::Union{NamedTuple, <:AbstractVector}) begin
     size = LuxCore.outputsize(model, x, LuxCore._default_rng())
     eltype = Real
 end