horde-ad-0.2.0.0: Higher Order Reverse Derivatives Efficiently - Automatic Differentiation
Safe HaskellNone
LanguageGHC2024

HordeAd.Core.AstEnv

Description

The environment datatype and operations for creating and accessing it.

Synopsis

Documentation

type AstEnv (target :: Target) = DEnumMap (AstVarName 'FullSpan) target Source #

The environment that keeps values assigned to variables during interpretation.

emptyEnv :: forall (target :: Target). AstEnv target Source #

showsPrecAstEnv :: forall (target :: Target). (forall (y :: TK). KnownSTK y => Show (target y)) => Int -> AstEnv target -> ShowS Source #

extendEnv :: forall target (s :: AstSpanType) (y :: TK). AstVarName s y -> target y -> AstEnv target -> AstEnv target Source #

extendEnvI :: forall (target :: Target). BaseTensor target => IntVarName -> IntOf target -> AstEnv target -> AstEnv target Source #

extendEnvVarsS :: forall (target :: Target) (sh :: [Nat]). BaseTensor target => AstVarListS sh -> IxSOf target sh -> AstEnv target -> AstEnv target Source #