NN.Backend.Torch.Codegen

Documentation

data TorchState

Constructors

TorchState

Fields

_statements :: [Stat]

_sequential :: Maybe String

_criteria :: [String]

_count :: Int

Instances

MonadState TorchState Torch

statements :: Lens' TorchState [Stat]

sequential :: Lens' TorchState (Maybe String)

criteria :: Lens' TorchState [String]

count :: Lens' TorchState Int

newtype Torch a

Constructors

Torch

Fields

_unTorch :: State TorchState a

Instances

Monad Torch

Functor Torch

Applicative Torch

MonadState TorchState Torch

initialize :: Torch ()

fresh :: String -> Torch String

insertModule :: Module Exp -> Torch ()

finalize :: Torch Block

runTorch :: [LayerParameter] -> Torch Block

lower :: [LayerParameter] -> Block

codegen :: Block -> String