Class Function

jvm/core/src/main/java/org/apache/tvm/Function.java:27–294

TVM Packed Function.

 * TVM Packed Function.
 */
public class Function extends TVMObject {
  /**
   * Get registered function.
   * @param name full function name.
   * @return TVM function.
   */
  public static Function getFunction(final String name) {
    return getGlobalFunc(name, true);
  }

  /**
   * Get list of global functions registered.
   * @return List of global functions names.
   */
  private static List<String> listGlobalFuncNames() {
    List<String> names = new ArrayList<String>();
    Base.checkCall(Base._LIB.tvmFFIFunctionListGlobalNames(names));
    return Collections.unmodifiableList(names);
  }

  /**
   * Get a global function by name.
   * @param name The name of the function.
   * @param allowMissing Whether allow missing function or raise an error.
   * @return The function to be returned, None if function is missing.
   */
  private static Function getGlobalFunc(String name, boolean allowMissing) {
    Base.RefLong handle = new Base.RefLong();
    Base.checkCall(Base._LIB.tvmFFIFunctionGetGlobal(name, handle));
    if (handle.value != 0) {
      return new Function(handle.value);
    } else {
      if (allowMissing) {
        return null;
      } else {
        throw new IllegalArgumentException("Cannot find global function " + name);
      }
    }
  }

  Function(long handle) {
    super(handle, TypeIndex.kTVMFFIFunction);
  }

  /**
   * Easy for user to get the instance from returned TVMValue.
   * @return this
   */
  @Override
  public Function asFunction() {
    return this;
  }

  /**
   * Invoke the function.
   * @return the result.
   */
  public TVMValue invoke() {