hasktorch-ffi-thc-0.0.1.0: Bindings to Cutorch

Safe HaskellNone
LanguageHaskell2010

Torch.FFI.THC.Short.TensorIndex

Synopsis

Documentation

c_indexCopy :: Ptr C'THCState -> Ptr C'THCudaShortTensor -> CInt -> Ptr C'THCudaLongTensor -> Ptr C'THCudaShortTensor -> IO () Source #

c_indexCopy : state res_ dim indices src -> void

c_indexAdd :: Ptr C'THCState -> Ptr C'THCudaShortTensor -> CInt -> Ptr C'THCudaLongTensor -> Ptr C'THCudaShortTensor -> IO () Source #

c_indexAdd : state res_ dim indices src -> void

c_indexFill :: Ptr C'THCState -> Ptr C'THCudaShortTensor -> CInt -> Ptr C'THCudaLongTensor -> CShort -> IO () Source #

c_indexFill : state tensor dim index val -> void

c_indexSelect :: Ptr C'THCState -> Ptr C'THCudaShortTensor -> Ptr C'THCudaShortTensor -> CInt -> Ptr C'THCudaLongTensor -> IO () Source #

c_indexSelect : state tensor src dim index -> void

c_take :: Ptr C'THCState -> Ptr C'THCudaShortTensor -> Ptr C'THCudaShortTensor -> Ptr C'THCudaLongTensor -> IO () Source #

c_take : state res_ src index -> void

c_put :: Ptr C'THCState -> Ptr C'THCudaShortTensor -> Ptr C'THCudaLongTensor -> Ptr C'THCudaShortTensor -> CInt -> IO () Source #

c_put : state res_ indices src accumulate -> void

c_indexCopy_long :: Ptr C'THCState -> Ptr C'THCudaShortTensor -> CInt -> Ptr C'THLongTensor -> Ptr C'THCudaShortTensor -> IO () Source #

c_indexCopy_long : state res_ dim indices src -> void

c_indexAdd_long :: Ptr C'THCState -> Ptr C'THCudaShortTensor -> CInt -> Ptr C'THLongTensor -> Ptr C'THCudaShortTensor -> IO () Source #

c_indexAdd_long : state res_ dim indices src -> void

c_indexFill_long :: Ptr C'THCState -> Ptr C'THCudaShortTensor -> CInt -> Ptr C'THLongTensor -> CShort -> IO () Source #

c_indexFill_long : state tensor dim index val -> void

c_indexSelect_long :: Ptr C'THCState -> Ptr C'THCudaShortTensor -> Ptr C'THCudaShortTensor -> CInt -> Ptr C'THLongTensor -> IO () Source #

c_indexSelect_long : state tensor src dim index -> void

c_calculateAdvancedIndexingOffsets :: Ptr C'THCState -> Ptr C'THCudaLongTensor -> Ptr C'THCudaShortTensor -> CPtrdiff -> Ptr (Ptr C'THCudaLongTensor) -> IO () Source #

c_calculateAdvancedIndexingOffsets : state output indexed baseOffset indexers -> void

p_indexCopy :: FunPtr (Ptr C'THCState -> Ptr C'THCudaShortTensor -> CInt -> Ptr C'THCudaLongTensor -> Ptr C'THCudaShortTensor -> IO ()) Source #

p_indexCopy : Pointer to function : state res_ dim indices src -> void

p_indexAdd :: FunPtr (Ptr C'THCState -> Ptr C'THCudaShortTensor -> CInt -> Ptr C'THCudaLongTensor -> Ptr C'THCudaShortTensor -> IO ()) Source #

p_indexAdd : Pointer to function : state res_ dim indices src -> void

p_indexFill :: FunPtr (Ptr C'THCState -> Ptr C'THCudaShortTensor -> CInt -> Ptr C'THCudaLongTensor -> CShort -> IO ()) Source #

p_indexFill : Pointer to function : state tensor dim index val -> void

p_indexSelect :: FunPtr (Ptr C'THCState -> Ptr C'THCudaShortTensor -> Ptr C'THCudaShortTensor -> CInt -> Ptr C'THCudaLongTensor -> IO ()) Source #

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

p_take :: FunPtr (Ptr C'THCState -> Ptr C'THCudaShortTensor -> Ptr C'THCudaShortTensor -> Ptr C'THCudaLongTensor -> IO ()) Source #

p_take : Pointer to function : state res_ src index -> void

p_put :: FunPtr (Ptr C'THCState -> Ptr C'THCudaShortTensor -> Ptr C'THCudaLongTensor -> Ptr C'THCudaShortTensor -> CInt -> IO ()) Source #

p_put : Pointer to function : state res_ indices src accumulate -> void

p_indexCopy_long :: FunPtr (Ptr C'THCState -> Ptr C'THCudaShortTensor -> CInt -> Ptr C'THLongTensor -> Ptr C'THCudaShortTensor -> IO ()) Source #

p_indexCopy_long : Pointer to function : state res_ dim indices src -> void

p_indexAdd_long :: FunPtr (Ptr C'THCState -> Ptr C'THCudaShortTensor -> CInt -> Ptr C'THLongTensor -> Ptr C'THCudaShortTensor -> IO ()) Source #

p_indexAdd_long : Pointer to function : state res_ dim indices src -> void

p_indexFill_long :: FunPtr (Ptr C'THCState -> Ptr C'THCudaShortTensor -> CInt -> Ptr C'THLongTensor -> CShort -> IO ()) Source #

p_indexFill_long : Pointer to function : state tensor dim index val -> void

p_indexSelect_long :: FunPtr (Ptr C'THCState -> Ptr C'THCudaShortTensor -> Ptr C'THCudaShortTensor -> CInt -> Ptr C'THLongTensor -> IO ()) Source #

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

p_calculateAdvancedIndexingOffsets :: FunPtr (Ptr C'THCState -> Ptr C'THCudaLongTensor -> Ptr C'THCudaShortTensor -> CPtrdiff -> Ptr (Ptr C'THCudaLongTensor) -> IO ()) Source #

p_calculateAdvancedIndexingOffsets : Pointer to function : state output indexed baseOffset indexers -> void