Struct tvm_rt::module::Module

pub struct Module(_);

Implementations

impl Module

pub fn default_fn(&mut self) -> Result<Function, Error>

pub fn get_function(
    &self,
    name: &str,
    query_import: bool
) -> Result<Function, Error>

Gets a function by name from a registered module.

pub fn import_module(&self, dependent_module: Module)

Imports a dependent module such as .ptx for cuda gpu.

pub fn load<P: AsRef<Path>>(path: &P) -> Result<Module, Error>

Loads a module shared library from path.

pub fn save_to_file(&self, name: String, fmt: String) -> Result<(), Error>

pub fn export_library(&self, name: String) -> Result<(), Error>

pub fn enabled(&self, target: &str) -> bool

Checks if a target device is enabled for a module.

pub unsafe fn handle(&self) -> TVMModuleHandle

Returns the underlying module handle.

Trait Implementations

impl Clone for Module

fn clone(&self) -> Module

Returns a copy of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.

impl Debug for Module

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter.

impl Deref for Module

type Target = ModuleNode

The resulting type after dereferencing.

fn deref(&self) -> &Self::Target

Dereferences the value.

impl<'a> From<&'a Module> for ArgValue<'a>

fn from(object_ref: &'a Module) -> ArgValue<'a>

Performs the conversion.

impl From<Module> for RetValue

fn from(object_ref: Module) -> RetValue

Performs the conversion.

impl From<ModuleNode> for Module

fn from(payload: ModuleNode) -> Self

Performs the conversion.

impl From<ObjectPtr<ModuleNode>> for Module

fn from(ptr: ObjectPtr<ModuleNode>) -> Self

Performs the conversion.

impl Hash for Module

fn hash<H: Hasher>(&self, state: &mut H)

Feeds this value into the given Hasher.

fn hash_slice<H>(data: &[Self], state: &mut H) where
    H: Hasher, 

Feeds a slice of this type into the given Hasher.

impl IsObjectRef for Module

type Object = ModuleNode

fn as_ptr(&self) -> Option<&ObjectPtr<Self::Object>>

fn into_ptr(self) -> Option<ObjectPtr<Self::Object>>

fn from_ptr(object_ptr: Option<ObjectPtr<Self::Object>>) -> Self

fn null() -> Self

fn into_arg_value<'a>(&'a self) -> ArgValue<'a>

fn from_arg_value<'a>(arg_value: ArgValue<'a>) -> Result<Self, Error>

fn into_ret_value<'a>(self) -> RetValue

fn from_ret_value<'a>(ret_value: RetValue) -> Result<Self, Error>

fn upcast<U>(self) -> U where
    U: IsObjectRef,
    Self::Object: AsRef<U::Object>, 

fn downcast<U>(self) -> Result<U, Error> where
    U: IsObjectRef,
    U::Object: AsRef<Self::Object>, 

impl PartialEq<Module> for Module

fn eq(&self, other: &Self) -> bool

This method tests for self and other values to be equal, and is used by ==.

fn ne(&self, other: &Rhs) -> bool

This method tests for !=.

impl<'a> TryFrom<ArgValue<'a>> for Module

type Error = Error

The type returned in the event of a conversion error.

fn try_from(arg_value: ArgValue<'a>) -> Result<Module>

Performs the conversion.

impl TryFrom<RetValue> for Module

type Error = Error

The type returned in the event of a conversion error.

fn try_from(ret_val: RetValue) -> Result<Module>

Performs the conversion.

impl Eq for Module