{-# LANGUAGE ForeignFunctionInterface #-} module Torch.FFI.THC.Short.TensorIndex where import Foreign import Foreign.C.Types import Data.Word import Data.Int import Torch.Types.TH import Torch.Types.THC -- | c_indexCopy : state res_ dim indices src -> void foreign import ccall "THCTensorIndex.h THCudaShortTensor_indexCopy" c_indexCopy :: Ptr C'THCState -> Ptr C'THCudaShortTensor -> CInt -> Ptr C'THCudaLongTensor -> Ptr C'THCudaShortTensor -> IO () -- | c_indexAdd : state res_ dim indices src -> void foreign import ccall "THCTensorIndex.h THCudaShortTensor_indexAdd" c_indexAdd :: Ptr C'THCState -> Ptr C'THCudaShortTensor -> CInt -> Ptr C'THCudaLongTensor -> Ptr C'THCudaShortTensor -> IO () -- | c_indexFill : state tensor dim index val -> void foreign import ccall "THCTensorIndex.h THCudaShortTensor_indexFill" c_indexFill :: Ptr C'THCState -> Ptr C'THCudaShortTensor -> CInt -> Ptr C'THCudaLongTensor -> CShort -> IO () -- | c_indexSelect : state tensor src dim index -> void foreign import ccall "THCTensorIndex.h THCudaShortTensor_indexSelect" c_indexSelect :: Ptr C'THCState -> Ptr C'THCudaShortTensor -> Ptr C'THCudaShortTensor -> CInt -> Ptr C'THCudaLongTensor -> IO () -- | c_take : state res_ src index -> void foreign import ccall "THCTensorIndex.h THCudaShortTensor_take" c_take :: Ptr C'THCState -> Ptr C'THCudaShortTensor -> Ptr C'THCudaShortTensor -> Ptr C'THCudaLongTensor -> IO () -- | c_put : state res_ indices src accumulate -> void foreign import ccall "THCTensorIndex.h THCudaShortTensor_put" c_put :: Ptr C'THCState -> Ptr C'THCudaShortTensor -> Ptr C'THCudaLongTensor -> Ptr C'THCudaShortTensor -> CInt -> IO () -- | c_indexCopy_long : state res_ dim indices src -> void foreign import ccall "THCTensorIndex.h THCudaShortTensor_indexCopy_long" c_indexCopy_long :: Ptr C'THCState -> Ptr C'THCudaShortTensor -> CInt -> Ptr C'THLongTensor -> Ptr C'THCudaShortTensor -> IO () -- | c_indexAdd_long : state res_ dim indices src -> void foreign import ccall "THCTensorIndex.h THCudaShortTensor_indexAdd_long" c_indexAdd_long :: Ptr C'THCState -> Ptr C'THCudaShortTensor -> CInt -> Ptr C'THLongTensor -> Ptr C'THCudaShortTensor -> IO () -- | c_indexFill_long : state tensor dim index val -> void foreign import ccall "THCTensorIndex.h THCudaShortTensor_indexFill_long" c_indexFill_long :: Ptr C'THCState -> Ptr C'THCudaShortTensor -> CInt -> Ptr C'THLongTensor -> CShort -> IO () -- | c_indexSelect_long : state tensor src dim index -> void foreign import ccall "THCTensorIndex.h THCudaShortTensor_indexSelect_long" c_indexSelect_long :: Ptr C'THCState -> Ptr C'THCudaShortTensor -> Ptr C'THCudaShortTensor -> CInt -> Ptr C'THLongTensor -> IO () -- | c_calculateAdvancedIndexingOffsets : state output indexed baseOffset indexers -> void foreign import ccall "THCTensorIndex.h THCudaShortTensor_calculateAdvancedIndexingOffsets" c_calculateAdvancedIndexingOffsets :: Ptr C'THCState -> Ptr C'THCudaLongTensor -> Ptr C'THCudaShortTensor -> CPtrdiff -> Ptr (Ptr C'THCudaLongTensor) -> IO () -- | p_indexCopy : Pointer to function : state res_ dim indices src -> void foreign import ccall "THCTensorIndex.h &THCudaShortTensor_indexCopy" p_indexCopy :: FunPtr (Ptr C'THCState -> Ptr C'THCudaShortTensor -> CInt -> Ptr C'THCudaLongTensor -> Ptr C'THCudaShortTensor -> IO ()) -- | p_indexAdd : Pointer to function : state res_ dim indices src -> void foreign import ccall "THCTensorIndex.h &THCudaShortTensor_indexAdd" p_indexAdd :: FunPtr (Ptr C'THCState -> Ptr C'THCudaShortTensor -> CInt -> Ptr C'THCudaLongTensor -> Ptr C'THCudaShortTensor -> IO ()) -- | p_indexFill : Pointer to function : state tensor dim index val -> void foreign import ccall "THCTensorIndex.h &THCudaShortTensor_indexFill" p_indexFill :: FunPtr (Ptr C'THCState -> Ptr C'THCudaShortTensor -> CInt -> Ptr C'THCudaLongTensor -> CShort -> IO ()) -- | p_indexSelect : Pointer to function : state tensor src dim index -> void foreign import ccall "THCTensorIndex.h &THCudaShortTensor_indexSelect" p_indexSelect :: FunPtr (Ptr C'THCState -> Ptr C'THCudaShortTensor -> Ptr C'THCudaShortTensor -> CInt -> Ptr C'THCudaLongTensor -> IO ()) -- | p_take : Pointer to function : state res_ src index -> void foreign import ccall "THCTensorIndex.h &THCudaShortTensor_take" p_take :: FunPtr (Ptr C'THCState -> Ptr C'THCudaShortTensor -> Ptr C'THCudaShortTensor -> Ptr C'THCudaLongTensor -> IO ()) -- | p_put : Pointer to function : state res_ indices src accumulate -> void foreign import ccall "THCTensorIndex.h &THCudaShortTensor_put" p_put :: FunPtr (Ptr C'THCState -> Ptr C'THCudaShortTensor -> Ptr C'THCudaLongTensor -> Ptr C'THCudaShortTensor -> CInt -> IO ()) -- | p_indexCopy_long : Pointer to function : state res_ dim indices src -> void foreign import ccall "THCTensorIndex.h &THCudaShortTensor_indexCopy_long" p_indexCopy_long :: FunPtr (Ptr C'THCState -> Ptr C'THCudaShortTensor -> CInt -> Ptr C'THLongTensor -> Ptr C'THCudaShortTensor -> IO ()) -- | p_indexAdd_long : Pointer to function : state res_ dim indices src -> void foreign import ccall "THCTensorIndex.h &THCudaShortTensor_indexAdd_long" p_indexAdd_long :: FunPtr (Ptr C'THCState -> Ptr C'THCudaShortTensor -> CInt -> Ptr C'THLongTensor -> Ptr C'THCudaShortTensor -> IO ()) -- | p_indexFill_long : Pointer to function : state tensor dim index val -> void foreign import ccall "THCTensorIndex.h &THCudaShortTensor_indexFill_long" p_indexFill_long :: FunPtr (Ptr C'THCState -> Ptr C'THCudaShortTensor -> CInt -> Ptr C'THLongTensor -> CShort -> IO ()) -- | p_indexSelect_long : Pointer to function : state tensor src dim index -> void foreign import ccall "THCTensorIndex.h &THCudaShortTensor_indexSelect_long" p_indexSelect_long :: FunPtr (Ptr C'THCState -> Ptr C'THCudaShortTensor -> Ptr C'THCudaShortTensor -> CInt -> Ptr C'THLongTensor -> IO ()) -- | p_calculateAdvancedIndexingOffsets : Pointer to function : state output indexed baseOffset indexers -> void foreign import ccall "THCTensorIndex.h &THCudaShortTensor_calculateAdvancedIndexingOffsets" p_calculateAdvancedIndexingOffsets :: FunPtr (Ptr C'THCState -> Ptr C'THCudaLongTensor -> Ptr C'THCudaShortTensor -> CPtrdiff -> Ptr (Ptr C'THCudaLongTensor) -> IO ())