Torch.FFI.Tests

Documentation

data TestFunctions state tensor real accreal

Constructors

TestFunctions

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 ())

type RealConstr n = (Num n, Show n, Eq n)

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

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