diff options
Diffstat (limited to 'src/AST.hs')
| -rw-r--r-- | src/AST.hs | 6 |
1 files changed, 3 insertions, 3 deletions
@@ -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 |