fix pb and pb2 for sparsepoolprod

ACEsuit · Jun 12, 2024 · 1e2b378 · 1e2b378
1 parent f919f42
commit 1e2b378
Show file tree

Hide file tree

Showing 3 changed files with 120 additions and 332 deletions.
diff --git a/src/Polynomials4ML.jl b/src/Polynomials4ML.jl
@@ -13,6 +13,7 @@ import WithAlloc: whatalloc
 using LuxCore, Random, StaticArrays
 import ChainRulesCore: rrule, frule, NoTangent, ZeroTangent
 using HyperDualNumbers: Hyper
+using ForwardDiff: Dual, extract_derivative
 
 import LuxCore: AbstractExplicitLayer, AbstractExplicitContainerLayer, 
                  initialparameters, initialstates                 
@@ -28,10 +29,10 @@ function index end
 function orthpolybasis end
 function degree end 
 
-function pullback end
 function pullback! end
 function pullback end
-function pullback! end
+function pullback2! end
+function pullback2 end
 function pushforward end
 function pushforward! end