}];

  let arguments = (ins
    MlrtPromiseType:$promise,
    TFTensorType:$tensor
  );

  let assemblyFormat = "operands attr-dict";
}

def PromiseFutureOp: TensorflowMlrt_Op<"promise_future", []> {
  let summary = "Set a tensor future in a promise";

  let description = [{
    Set a tensor future in a promise.

  // CHECK: [[y_future:%.*]] = tf_mlrt.async_executeop
  %y = "tf.TestAsyncIdentity"(%x) {__op_key = 0: i32, T = i32} : (tensor<i32>) -> tensor<i32>
  // CHECK: tf_mlrt.promise_future
  // CHECK-SAME: [[y_future]]
  "tf_mlrt.tf_promise"(%p, %y): (!mlrt.promise, tensor<i32>) -> ()
  return
}

// CHECK-LABEL: @main
// CHECK-SAME: ([[input0:%.*]]: !tf_mlrt.tensor)