Safe Haskell	None
Language	Haskell2010

Torch.FFI.Tests

Documentation

data TestFunctions state tensor real accreal Source #

Constructors

Fields

_new :: state -> IO tensor
_newWithSize1d :: state -> CLLong -> IO tensor
_newWithSize2d :: state -> CLLong -> CLLong -> IO tensor
_newWithSize3d :: state -> CLLong -> CLLong -> CLLong -> IO tensor
_newWithSize4d :: state -> CLLong -> CLLong -> CLLong -> CLLong -> IO tensor
_nDimension :: state -> tensor -> IO CInt
_set1d :: state -> tensor -> CLLong -> real -> IO ()
_get1d :: state -> tensor -> CLLong -> IO real
_set2d :: state -> tensor -> CLLong -> CLLong -> real -> IO ()
_get2d :: state -> tensor -> CLLong -> CLLong -> IO real
_set3d :: state -> tensor -> CLLong -> CLLong -> CLLong -> real -> IO ()
_get3d :: state -> tensor -> CLLong -> CLLong -> CLLong -> IO real
_set4d :: state -> tensor -> CLLong -> CLLong -> CLLong -> CLLong -> real -> IO ()
_get4d :: state -> tensor -> CLLong -> CLLong -> CLLong -> CLLong -> IO real
_size :: state -> tensor -> CInt -> IO CLLong
_fill :: state -> tensor -> real -> IO ()
_free :: state -> tensor -> IO ()
_sumall :: state -> tensor -> IO accreal
_prodall :: state -> tensor -> IO accreal
_zero :: state -> tensor -> IO ()
_dot :: Maybe (state -> tensor -> tensor -> IO accreal)
_abs :: Maybe (state -> tensor -> tensor -> IO ())

signedSuite :: (RealConstr real, RealConstr accreal) => state -> TestFunctions state tensor real accreal -> Spec Source #

dotSpec :: (Show a, Eq a, Num real, Num a) => state -> TestFunctions state tensor real accreal -> (state -> tensor -> tensor -> IO a) -> SpecM () () Source #