Torch.Backend

Documentation

data Backend

Constructors

CPU

CUDA

HIP

SparseCPU

SparseCUDA

XLA

MPS

Instances

Show Backend

Defined in Torch.Backend

Methods

showsPrec :: Int -> Backend -> ShowS

show :: Backend -> String

showList :: [Backend] -> ShowS

Eq Backend

Defined in Torch.Backend

Methods

(==) :: Backend -> Backend -> Bool

(/=) :: Backend -> Backend -> Bool

Castable Backend Backend

Defined in Torch.Backend

Methods

cast :: Backend -> (Backend -> IO r) -> IO r

uncast :: Backend -> (Backend -> IO r) -> IO r