| Safe Haskell | Safe-Inferred |
|---|---|
| Language | Haskell2010 |
Torch.Internal.Managed.Type.Module
Documentation
newModule :: ForeignPtr StdString -> IO (ForeignPtr Module) Source #
forward :: ForeignPtr Module -> ForeignPtr (StdVector IValue) -> IO (ForeignPtr IValue) Source #
registerParameter :: ForeignPtr Module -> ForeignPtr StdString -> ForeignPtr Tensor -> CBool -> IO () Source #
registerModule :: ForeignPtr Module -> ForeignPtr StdString -> ForeignPtr Module -> IO () Source #
runMethod :: ForeignPtr Module -> ForeignPtr StdString -> ForeignPtr (C10List IValue) -> IO (Ptr IValue) Source #
runMethod1 :: ForeignPtr Module -> ForeignPtr StdString -> ForeignPtr IValue -> IO (Ptr IValue) Source #
getParameters :: ForeignPtr Module -> IO (ForeignPtr TensorList) Source #
setParameters :: ForeignPtr Module -> ForeignPtr TensorList -> IO () Source #
getNamedParameters :: ForeignPtr Module -> IO [(ForeignPtr StdString, ForeignPtr Tensor)] Source #
getNamedBuffers :: ForeignPtr Module -> IO [(ForeignPtr StdString, ForeignPtr Tensor)] Source #
getNamedAttributes :: ForeignPtr Module -> IO [(ForeignPtr StdString, ForeignPtr IValue)] Source #
getNamedModules :: ForeignPtr Module -> IO [(ForeignPtr StdString, ForeignPtr Module)] Source #
getNamedChildren :: ForeignPtr Module -> IO [(ForeignPtr StdString, ForeignPtr Module)] Source #
toDevice :: ForeignPtr Module -> DeviceType -> Int16 -> IO () Source #
clone :: ForeignPtr Module -> IO (ForeignPtr Module) Source #
define :: ForeignPtr Module -> ForeignPtr StdString -> IO () Source #
trace :: String -> String -> (ForeignPtr TensorList -> IO (ForeignPtr TensorList)) -> ForeignPtr TensorList -> IO (ForeignPtr Module) Source #
traceAsGraph :: (ForeignPtr TensorList -> IO (ForeignPtr TensorList)) -> ForeignPtr TensorList -> IO (ForeignPtr (SharedPtr JitGraph)) Source #
printGraph :: ForeignPtr (SharedPtr JitGraph) -> IO (ForeignPtr StdString) Source #
printOnnx :: ForeignPtr (SharedPtr JitGraph) -> IO (ForeignPtr StdString) Source #