hasktorch-ffi-thc-0.0.1.0: Bindings to Cutorch

Safe HaskellNone
LanguageHaskell2010

Torch.FFI.THC.Short.TensorTopK

Synopsis

Documentation

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

c_topk : state topK indices input k dim dir sorted -> void

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

p_topk : Pointer to function : state topK indices input k dim dir sorted -> void