Method pushArgToStack

jvm/core/src/main/java/org/apache/tvm/Function.java:206–248

(Object arg)

  }

  private static void pushArgToStack(Object arg) {
    if (arg instanceof TensorBase) {
      TensorBase nd = (TensorBase) arg;
      if (nd instanceof Tensor) {
        Base._LIB.tvmFFIFunctionPushArgHandle(((Tensor) nd).handle, TypeIndex.kTVMFFITensor);
      } else {
        Base._LIB.tvmFFIFunctionPushArgHandle(nd.dltensorHandle, TypeIndex.kTVMFFIDLTensorPtr);
      }
    } else if (arg instanceof TVMObject) {
      TVMObject obj = (TVMObject) arg;
      Base._LIB.tvmFFIFunctionPushArgHandle(obj.handle, obj.typeIndex);
    } else if (arg instanceof Integer) {
      Base._LIB.tvmFFIFunctionPushArgLong((Integer) arg);
    } else if (arg instanceof Long) {
      Base._LIB.tvmFFIFunctionPushArgLong((Long) arg);
    } else if (arg instanceof Float) {
      Base._LIB.tvmFFIFunctionPushArgDouble((Float) arg);
    } else if (arg instanceof Double) {
      Base._LIB.tvmFFIFunctionPushArgDouble((Double) arg);
    } else if (arg instanceof String) {
      Base._LIB.tvmFFIFunctionPushArgString((String) arg);
    } else if (arg instanceof byte[]) {
      Base._LIB.tvmFFIFunctionPushArgBytes((byte[]) arg);
    } else if (arg instanceof Device) {
      Base._LIB.tvmFFIFunctionPushArgDevice((Device) arg);
    } else if (arg instanceof TVMValueBytes) {
      byte[] bytes = ((TVMValueBytes) arg).value;
      Base._LIB.tvmFFIFunctionPushArgBytes(bytes);
    } else if (arg instanceof TVMValueString) {
      String str = ((TVMValueString) arg).value;
      Base._LIB.tvmFFIFunctionPushArgString(str);
    } else if (arg instanceof TVMValueDouble) {
      double value = ((TVMValueDouble) arg).value;
      Base._LIB.tvmFFIFunctionPushArgDouble(value);
    } else if (arg instanceof TVMValueLong) {
      long value = ((TVMValueLong) arg).value;
      Base._LIB.tvmFFIFunctionPushArgLong(value);
    } else if (arg instanceof TVMValueNull) {
      Base._LIB.tvmFFIFunctionPushArgHandle(0, TypeIndex.kTVMFFINone);
    } else {
      throw new IllegalArgumentException("Invalid argument: " + arg);
    }
  }

  public static interface Callback {
    public Object invoke(TVMValue... args);