packages feed

libtorch-ffi-2.0.0.0: src/Torch/Internal/Managed/Serialize.hs

module Torch.Internal.Managed.Serialize where

import Foreign.ForeignPtr

import qualified Torch.Internal.Unmanaged.Serialize as Unmanaged
import Torch.Internal.Type
import Torch.Internal.Class
import Torch.Internal.Cast
import Torch.Internal.Objects


save :: ForeignPtr TensorList -> FilePath -> IO ()
save = _cast2 Unmanaged.save

load :: FilePath -> IO (ForeignPtr TensorList)
load = _cast1 Unmanaged.load

pickleSave :: ForeignPtr IValue -> FilePath -> IO ()
pickleSave = _cast2 Unmanaged.pickleSave

pickleLoad :: FilePath -> IO (ForeignPtr IValue)
pickleLoad = _cast1 Unmanaged.pickleLoad