hasktorch-ffi-thc-0.0.1.0: Bindings to Cutorch

Safe HaskellNone
LanguageHaskell2010

Torch.FFI.THC.Int.TensorScatterGather

Synopsis

Documentation

c_gather :: Ptr C'THCState -> Ptr C'THCudaIntTensor -> Ptr C'THCudaIntTensor -> CInt -> Ptr C'THCudaLongTensor -> IO () Source #

c_gather : state tensor src dim index -> void

c_scatter :: Ptr C'THCState -> Ptr C'THCudaIntTensor -> CInt -> Ptr C'THCudaLongTensor -> Ptr C'THCudaIntTensor -> IO () Source #

c_scatter : state tensor dim index src -> void

c_scatterAdd :: Ptr C'THCState -> Ptr C'THCudaIntTensor -> CInt -> Ptr C'THCudaLongTensor -> Ptr C'THCudaIntTensor -> IO () Source #

c_scatterAdd : state tensor dim index src -> void

c_scatterFill :: Ptr C'THCState -> Ptr C'THCudaIntTensor -> CInt -> Ptr C'THCudaLongTensor -> CInt -> IO () Source #

c_scatterFill : state tensor dim index value -> void

p_gather :: FunPtr (Ptr C'THCState -> Ptr C'THCudaIntTensor -> Ptr C'THCudaIntTensor -> CInt -> Ptr C'THCudaLongTensor -> IO ()) Source #

p_gather : Pointer to function : state tensor src dim index -> void

p_scatter :: FunPtr (Ptr C'THCState -> Ptr C'THCudaIntTensor -> CInt -> Ptr C'THCudaLongTensor -> Ptr C'THCudaIntTensor -> IO ()) Source #

p_scatter : Pointer to function : state tensor dim index src -> void

p_scatterAdd :: FunPtr (Ptr C'THCState -> Ptr C'THCudaIntTensor -> CInt -> Ptr C'THCudaLongTensor -> Ptr C'THCudaIntTensor -> IO ()) Source #

p_scatterAdd : Pointer to function : state tensor dim index src -> void

p_scatterFill :: FunPtr (Ptr C'THCState -> Ptr C'THCudaIntTensor -> CInt -> Ptr C'THCudaLongTensor -> CInt -> IO ()) Source #

p_scatterFill : Pointer to function : state tensor dim index value -> void