hasktorch-0.2.1.8: Haskell bindings to libtorch, supporting both typed and untyped tensors.

Safe Haskell	None
Language	Haskell2010

Torch.Device

Documentation

data DeviceType Source #

Constructors

Instances

Instances details

Show DeviceType Source #

Instance details

Defined in Torch.Device

Methods

showsPrec :: Int -> DeviceType -> ShowS #

show :: DeviceType -> String #

showList :: [DeviceType] -> ShowS #

Eq DeviceType Source #

Instance details

Defined in Torch.Device

Methods

(==) :: DeviceType -> DeviceType -> Bool #

(/=) :: DeviceType -> DeviceType -> Bool #

Ord DeviceType Source #

Instance details

Defined in Torch.Device

Methods

compare :: DeviceType -> DeviceType -> Ordering #

(<) :: DeviceType -> DeviceType -> Bool #

(<=) :: DeviceType -> DeviceType -> Bool #

(>) :: DeviceType -> DeviceType -> Bool #

(>=) :: DeviceType -> DeviceType -> Bool #

max :: DeviceType -> DeviceType -> DeviceType #

min :: DeviceType -> DeviceType -> DeviceType #

DeviceType DeviceType Source #

Instance details

Defined in Torch.Device

Methods

cast :: DeviceType -> (DeviceType -> IO r) -> IO r #

uncast :: DeviceType -> (DeviceType -> IO r) -> IO r #

HasToDevice device' device (HList xs) (HList ys) => ReduceGradients device' '[device] ('[HList xs] :: [Type]) (ys :: [k]) Source #

Instance details

Defined in Torch.Typed.NN.DataParallel

Methods

reduceGradients :: HList '[HList xs] -> HList ys Source #

(HasToDevice device' device (HList xs) (HList ys), ReduceGradients device' devices xxs ys, HZipWith SumF ys ys ys, 1 <= ListLength xxs) => ReduceGradients device' (device ': devices) (HList xs ': xxs :: [Type]) (ys :: [k]) Source #

Instance details

Defined in Torch.Typed.NN.DataParallel

Methods

reduceGradients :: HList (HList xs ': xxs) -> HList ys Source #

HasReplicate ('[] :: [(DeviceType, Nat)]) device f ('[] :: [Type]) Source #

Instance details

Defined in Torch.Typed.Device

Methods

replicate :: f -> HList ('[] :: [Type]) Source #

HasToDevices ('[] :: [(DeviceType, Nat)]) ('[] :: [(DeviceType, Nat)]) ('[] :: [Type]) ('[] :: [Type]) Source #

Instance details

Defined in Torch.Typed.Device

Methods

toDevices :: HList ('[] :: [Type]) -> HList ('[] :: [Type]) Source #

(chunks ~ ListLength devices', tensorChunks ~ (Chunk chunks 0 shape dtype device :: [Type]), Castable (HList tensorChunks) [ATenTensor], devices ~ HReplicateR chunks device, HasToDevices devices' devices tensorChunks gs, KnownNat chunks) => HasScatter (devices' :: [(DeviceType, Nat)]) (device :: (DeviceType, Nat)) (Tensor device dtype shape) (gs :: [Type]) Source #

Instance details

Defined in Torch.Typed.Device

Methods

scatter :: Tensor device dtype shape -> HList gs Source #

(HZipWithM Concurrently GradConcurrentlyF parameters losses gradients', ReduceGradients device' devices gradients' gradients) => HasGradConcurrently (device' :: (DeviceType, Nat)) (devices :: [(DeviceType, Nat)]) (parameters :: [k1]) (losses :: [k1]) (gradients :: [k2]) Source #

Instance details

Defined in Torch.Typed.NN.DataParallel

Methods

gradConcurrently :: HList parameters -> HList losses -> Concurrently (HList gradients) Source #

(chunks ~ ListLength fs, devices ~ GetDevices fs, devices' ~ HReplicateR chunks device', HasToDevices devices' devices fs tensorChunks, '(shape, dtype, device') ~ Cat 0 tensorChunks, Castable (HList tensorChunks) [ATenTensor]) => HasGather (device' :: (DeviceType, Nat)) (devices :: [(DeviceType, Nat)]) (fs :: [Type]) (Tensor device' dtype shape) Source #

Instance details

Defined in Torch.Typed.Device

Methods

gather :: HList fs -> Tensor device' dtype shape Source #

(HasReplicate devices' device f gs, HasToDevice device' device f g) => HasReplicate (device' ': devices') device f (g ': gs) Source #

Instance details

Defined in Torch.Typed.Device

Methods

replicate :: f -> HList (g ': gs) Source #

(HasToDevices devices' devices fs gs, HasToDevice device' device f g) => HasToDevices (device' ': devices') (device ': devices) (f ': fs) (g ': gs) Source #

Instance details

Defined in Torch.Typed.Device

Methods

toDevices :: HList (f ': fs) -> HList (g ': gs) Source #

KnownNat n => KnownDevice '('CPU, n) Source #

Instance details

Defined in Torch.Typed.Tensor

Methods

deviceVal :: Device Source #

KnownNat n => KnownDevice '('CUDA, n) Source #

Instance details

Defined in Torch.Typed.Tensor

Methods

deviceVal :: Device Source #

KnownNat n => KnownDevice '('MPS, n) Source #

Instance details

Defined in Torch.Typed.Tensor

Methods

deviceVal :: Device Source #

data Device Source #

Constructors

Fields

deviceType :: DeviceType
deviceIndex :: Int16

Instances

Instances details

Show Device Source #

Instance details

Defined in Torch.Device

Methods

showsPrec :: Int -> Device -> ShowS #

show :: Device -> String #

showList :: [Device] -> ShowS #

Eq Device Source #

Instance details

Defined in Torch.Device

Methods

(==) :: Device -> Device -> Bool #

(/=) :: Device -> Device -> Bool #

Ord Device Source #

Instance details

Defined in Torch.Device

Methods

compare :: Device -> Device -> Ordering #

(<) :: Device -> Device -> Bool #

(<=) :: Device -> Device -> Bool #

(>) :: Device -> Device -> Bool #

(>=) :: Device -> Device -> Bool #

max :: Device -> Device -> Device #

min :: Device -> Device -> Device #

Show (TVar (Either (Word64, Device) (ForeignPtr Generator))) Source #

Instance details

Defined in Torch.Random

Methods

showsPrec :: Int -> TVar (Either (Word64, Device) (ForeignPtr Generator)) -> ShowS #

show :: TVar (Either (Word64, Device) (ForeignPtr Generator)) -> String #

showList :: [TVar (Either (Word64, Device) (ForeignPtr Generator))] -> ShowS #