包 org.apache.tvm

类 Function

java.lang.Object
org.apache.tvm.TVMValue
org.apache.tvm.Function
public class Function extends TVMValue
TVM 打包函数。

  • 字段详细信息

    • isResident

      public final boolean isResident

  • 方法详细信息

    • getFunction

      public static Function getFunction(String name)
      获取已注册的函数。
      参数
      name - 完整函数名称。
      返回：
      TVM 函数。

    • finalize

      protected void finalize() throws Throwable
      覆盖：
      finalize in class Object
      抛出：
      Throwable

    • asFunction

      public Function asFunction()
      方便用户从返回的 TVMValue 中获取实例。
      覆盖：
      asFunction in class TVMValue
      返回：
      this

    • release

      public void release()
      释放 Function。

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

      覆盖：
      release in class TVMValue

    • invoke

      public TVMValue invoke()
      调用该函数。
      返回：
      结果。

    • pushArg

      public Function pushArg(int arg)
      将参数推送到函数。
      参数
      arg - int 参数。
      返回：
      this

    • pushArg

      public Function pushArg(long arg)
      将参数推送到函数。
      参数
      arg - long 参数。
      返回：
      this

    • pushArg

      public Function pushArg(float arg)
      将参数推送到函数。
      参数
      arg - float 参数。
      返回：
      this

    • pushArg

      public Function pushArg(double arg)
      将参数推送到函数。
      参数
      arg - double 参数。
      返回：
      this

    • pushArg

      public Function pushArg(String arg)
      将参数推送到函数。
      参数
      arg - String 参数。
      返回：
      this

    • pushArg

      public Function pushArg(NDArrayBase arg)
      将参数推送到函数。
      参数
      arg - NDArray。
      返回：
      this

    • pushArg

      public Function pushArg(Module arg)
      将参数推送到函数。
      参数
      arg - Module。
      返回：
      this

    • pushArg

      public Function pushArg(Function arg)
      将参数推送到函数。
      参数
      arg - Function。
      返回：
      this

    • pushArg

      public Function pushArg(byte[] arg)
      将参数推送到函数。
      参数
      arg - bytes。
      返回：
      this

    • pushArg

      public Function pushArg(Device arg)
      将参数推送到函数。
      参数
      arg - Device。
      返回：
      this

    • call

      public TVMValue call(Object... args)
      使用参数调用函数。
      参数
      args - 可以是 Integer、Long、Float、Double、String、NDArray。
      返回：
      结果。

    • register

      public static void register(String name, Function.Callback function, boolean override)
      注册用户定义的全局函数。
      参数
      name - 函数名称。
      function - 要注册的函数。
      override - 是否覆盖现有条目。

    • register

      public static void register(String name, Function.Callback function)
      注册用户定义的全局函数，不要覆盖现有条目。
      参数
      name - 函数名称。
      function - 要注册的函数。

    • convertFunc

      public static Function convertFunc(Function.Callback function)
      将 Java 函数转换为 TVM 函数。
      参数
      function - Java 函数。
      返回：
      TVM 函数。