diff options
| author | Tom Smeding <tom@tomsmeding.com> | 2024-10-22 22:02:06 +0200 |
|---|---|---|
| committer | Tom Smeding <tom@tomsmeding.com> | 2024-10-22 22:02:06 +0200 |
| commit | 79e072eddf0ec2a97ca455c27cb5ff6f2132bbab (patch) | |
| tree | 2099dc7e9d9a1109d844bca73277ca82983a02c2 /src/Language.hs | |
| parent | e7d7ac0fd8b81c1d6fae9ab7c1e4654133c631ea (diff) | |
Differentiate Replicate
Diffstat (limited to 'src/Language.hs')
| -rw-r--r-- | src/Language.hs | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/src/Language.hs b/src/Language.hs index 80de713..c2b844e 100644 --- a/src/Language.hs +++ b/src/Language.hs @@ -63,8 +63,8 @@ build1 a (v :-> b) = NEBuild1 a v b build :: SNat n -> NExpr env (Tup (Replicate n TIx)) -> (Var name (Tup (Replicate n TIx)) :-> NExpr ('(name, Tup (Replicate n TIx)) : env) t) -> NExpr env (TArr n t) build n a (v :-> b) = NEBuild n a v b -fold1i :: (Var name1 t :-> Var name2 t :-> NExpr ('(name2, t) : '(name1, t) : env) t) -> NExpr env (TArr (S n) t) -> NExpr env (TArr n t) -fold1i (v1 :-> v2 :-> e1) e2 = NEFold1Inner v1 v2 e1 e2 +fold1i :: (Var name1 t :-> Var name2 t :-> NExpr ('(name2, t) : '(name1, t) : env) t) -> NExpr env t -> NExpr env (TArr (S n) t) -> NExpr env (TArr n t) +fold1i (v1 :-> v2 :-> e1) e2 e3 = NEFold1Inner v1 v2 e1 e2 e3 sum1i :: ScalIsNumeric t ~ True => NExpr env (TArr (S n) (TScal t)) -> NExpr env (TArr n (TScal t)) sum1i e = NESum1Inner e |