summaryrefslogtreecommitdiff
path: root/src/AST.hs
diff options
context:
space:
mode:
authorTom Smeding <tom@tomsmeding.com>2024-11-09 22:59:12 +0100
committerTom Smeding <tom@tomsmeding.com>2024-11-09 22:59:12 +0100
commitd4d4473ee229674f73929c0860a7e29302330361 (patch)
tree65efd81ff635bad0e85cc518dc67c7c204313c73 /src/AST.hs
parent34887168c0e2deb549e0e7c77e837ab269d894a2 (diff)
Cleanup, more Language operations
Diffstat (limited to 'src/AST.hs')
-rw-r--r--src/AST.hs6
1 files changed, 3 insertions, 3 deletions
diff --git a/src/AST.hs b/src/AST.hs
index e3da634..263b806 100644
--- a/src/AST.hs
+++ b/src/AST.hs
@@ -104,9 +104,9 @@ data Expr x env t where
-- ECustom does not allow a derivative to be generated for 'a', and hence
-- none is propagated.
ECustom :: x t -> STy a -> STy b -> STy tape
- -> Expr x '[b, a] t -- ^ regular operation
- -> Expr x '[D1 b, D1 a] (TPair (D1 t) tape) -- ^ CHAD forward pass
- -> Expr x '[D2 t, tape] (D2 b) -- ^ CHAD reverse derivative
+ -> Expr x [b, a] t -- ^ regular operation
+ -> Expr x [D1 b, D1 a] (TPair (D1 t) tape) -- ^ CHAD forward pass
+ -> Expr x [D2 t, tape] (D2 b) -- ^ CHAD reverse derivative
-> Expr x env a -> Expr x env b
-> Expr x env t