hasktorch-0.2.0.0: Functional differentiable programming in Haskell
Safe HaskellSafe-Inferred
LanguageHaskell2010

Torch.Backend

Documentation

data Backend Source #

Constructors

CPU 
CUDA 
HIP 
SparseCPU 
SparseCUDA 
XLA 

Instances

Instances details
Show Backend Source # 
Instance details

Defined in Torch.Backend

Eq Backend Source # 
Instance details

Defined in Torch.Backend

Castable Backend Backend Source # 
Instance details

Defined in Torch.Backend

Methods

cast :: Backend -> (Backend0 -> IO r) -> IO r Source #

uncast :: Backend0 -> (Backend -> IO r) -> IO r Source #