caffegraph-0.1.0.2

Safe Haskell	None
Language	Haskell2010

NN.Backend.Torch.Torch

Documentation

data Module a Source

Constructors

Criterion a
Inner a

Instances

Functor Module
Show a => Show (Module a)

data TorchModule Source

Constructors

TorchModule Name Name [Exp]

Instances

Show TorchModule

torchExp :: TorchModule -> Exp Source

torchModules :: LayerParameter -> [Module TorchModule] Source

torchLayers :: [LayerTy] Source

isSequential :: Net -> Bool Source

clean :: Net -> Net Source

linearize :: Net -> Maybe [LayerParameter] Source