包 org.apache.tvm

类 Module

java.lang.Object
org.apache.tvm.TVMValue
org.apache.tvm.Module
public class Module extends TVMValue
TVM 编译函数的容器。

  • 字段详细信息

    • handle

      public final long handle

  • 方法详细信息

    • finalize

      protected void finalize() throws Throwable
      覆盖
      finalize 在类 Object
      抛出
      Throwable

    • asModule

      public Module asModule()
      方便用户从返回的 TVMValue 中获取实例。
      覆盖
      asModule 在类 TVMValue
      返回
      this

    • release

      public void release()
      释放模块。

      我们强烈建议您手动执行此操作，因为 GC 策略是延迟的。

      覆盖
      release 在类 TVMValue

    • entryFunc

      public Function entryFunc()
      获取入口函数。
      返回
      如果存在，则返回入口函数

    • getFunction

      public Function getFunction(String name, boolean queryImports)
      从模块中获取函数。
      参数
      name - 函数的名称。
      queryImports - 是否同时查询由此模块导入的模块。
      返回
      结果函数。

    • getFunction

      public Function getFunction(String name)

    • importModule

      public void importModule(Module module)
      将模块添加到当前模块的导入列表中。
      参数
      module - 另一个模块。

    • typeKey

      public String typeKey()
      获取模块的类型键。
      返回
      模块的类型键。

    • load

      public static Module load(String path, String fmt)
      从文件加载模块。
      参数
      path - 模块文件的路径。
      fmt - 文件格式，如果未指定，则将从文件后缀推断。
      返回
      加载的模块

    • load

      public static Module load(String path)

    • enabled

      public static boolean enabled(String target)
      模块运行时是否为目标启用，例如，以下代码检查是否启用了 CUDA。Module.enabled("cuda")
      参数
      target - 目标设备类型。
      返回
      运行时是否已启用。