caffegraph-0.1.0.2

Safe HaskellNone
LanguageHaskell2010

NN.Backend.Torch.Torch

Documentation

data Module a Source

Constructors

Criterion a 
Inner a 

Instances

data TorchModule Source

Constructors

TorchModule Name Name [Exp] 

Instances

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