Documentation

newModule :: Ptr StdString -> IO (Ptr Module) Source #

save :: Ptr Module -> FilePath -> IO () Source #

load :: FilePath -> IO (Ptr Module) Source #

forward :: Ptr Module -> Ptr (StdVector IValue) -> IO (Ptr IValue) Source #

registerParameter :: Ptr Module -> Ptr StdString -> Ptr Tensor -> CBool -> IO () Source #

registerModule :: Ptr Module -> Ptr StdString -> Ptr Module -> IO () Source #

train :: Ptr Module -> CBool -> IO () Source #

runMethod :: Ptr Module -> Ptr StdString -> Ptr (C10List IValue) -> IO (Ptr IValue) Source #

runMethod1 :: Ptr Module -> Ptr StdString -> Ptr IValue -> IO (Ptr IValue) Source #

getParameters :: Ptr Module -> IO (Ptr TensorList) Source #

setParameters :: Ptr Module -> Ptr TensorList -> IO () Source #

getNamedParameters :: Ptr Module -> IO [(Ptr StdString, Ptr Tensor)] Source #

getNamedBuffers :: Ptr Module -> IO [(Ptr StdString, Ptr Tensor)] Source #

getNamedAttributes :: Ptr Module -> IO [(Ptr StdString, Ptr IValue)] Source #

getNamedModules :: Ptr Module -> IO [(Ptr StdString, Ptr Module)] Source #

getNamedChildren :: Ptr Module -> IO [(Ptr StdString, Ptr Module)] Source #

toDevice :: Ptr Module -> DeviceType -> Int16 -> IO () Source #

clone :: Ptr Module -> IO (Ptr Module) Source #

define :: Ptr Module -> Ptr StdString -> IO () Source #

trace :: CString -> CString -> (Ptr TensorList -> IO (Ptr TensorList)) -> Ptr TensorList -> IO (Ptr Module) Source #

traceAsGraph :: (Ptr TensorList -> IO (Ptr TensorList)) -> Ptr TensorList -> IO (Ptr (SharedPtr JitGraph)) Source #

withJitGraph :: Ptr (SharedPtr JitGraph) -> (Ptr JitGraph -> IO a) -> IO a Source #

graphOutputs :: Ptr JitGraph -> IO [Ptr JitValue] Source #

graphInputs :: Ptr JitGraph -> IO [Ptr JitValue] Source #

graphNodes :: Ptr JitGraph -> IO [Ptr JitNode] Source #

nodeInputs :: Ptr JitNode -> IO [Ptr JitValue] Source #

nodeOutputs :: Ptr JitNode -> IO [Ptr JitValue] Source #

nodeKind :: Ptr JitNode -> IO (Ptr StdString) Source #

valueId :: Ptr JitValue -> IO CInt Source #

valueType :: Ptr JitValue -> IO (Ptr StdString) Source #

printGraph :: Ptr (SharedPtr JitGraph) -> IO (Ptr StdString) Source #

printOnnx :: Ptr (SharedPtr JitGraph) -> IO (Ptr StdString) Source #

dumpToStr :: Ptr Module -> CBool -> CBool -> CBool -> IO (Ptr StdString) Source #