// Store the control_token output for use by downstream nodes.
  maybe_control_node->second.outgoing = ctrl_op.getControl();

  // Remove the original op.
  op->replaceAllUsesWith(ctrl_op.getOutputs());
  op->erase();
  return ctrl_op.getOutputs();
}

// Build a FuncOp from a tflite SubGraph
// The buffers are directly taken
// from the deserialized flatbuffer as we do not have the type information to

    }

    @Internal
    @Override
    public TaskInputsInternal getInputs() {
        return taskInputs;
    }

    @Internal
    @Override
    public TaskOutputsInternal getOutputs() {
        return taskOutputs;
    }

    @Internal
    @Override
    public TaskDestroyables getDestroyables() {
        return taskDestroyables;
    }

    @Internal
    @Override

    send_tokens.push_back(send_token);
  }
  token = CreateSinkToken(builder, loc, send_tokens, token);

  SmallVector<Value, 4> recv_tokens;
  for (auto result : llvm::enumerate(host_compute.getOutputs())) {
    auto recv_token = CreateRecvOp(
        builder, loc, result.value(), host_compute.getRecvKey(), result.index(),
        token, xla::kXlaHostTransferTfRendezvousHandlerName, manual_sharding);

    @Internal
    TaskInputs getInputs();

    /**
     * <p>Returns the outputs of this task.</p>
     *
     * @return The outputs. Never returns null.
     */
    @Internal
    TaskOutputs getOutputs();

    /**
     * <p>Returns the destroyables of this task.</p>
     * @return The destroyables.  Never returns null.
     *
     * @since 4.0
     */
    @Internal

* link:{groovyDslPath}/org.gradle.api.Task.html#org.gradle.api.Task:outputs[Task.getOutputs()] of type link:{javadocPath}/org/gradle/api/tasks/TaskOutputs.html[TaskOutputs]

                                PatternRewriter &rewriter) const override {
    SmallVector<Type> result_types;
    result_types.reserve(op.getOutputs().size() + 1);
    for (const auto &output : op.getOutputs()) {
      Type ty = output.getType();
      if (auto tensor_ty = mlir::dyn_cast<RankedTensorType>(ty)) {
        if (!tensor_ty.hasStaticShape()) return failure();
      }

Method <org.gradle.api.DefaultTask.getOutputs()> has arguments/return type org.gradle.api.internal.TaskOutputsInternal that is not Gradle public API or primitive or built-in JDK classes or Kotlin classes in (DefaultTask.java:0)

    // Replace the ControlNodeOp with the wrapped operation.
    rewriter.setInsertionPointAfter(outer_op);
    auto* cloned_inner = rewriter.clone(*inner_op);
    for (auto it :
         llvm::zip(control_node_op.getOutputs(), cloned_inner->getResults())) {
      std::get<0>(it).replaceAllUsesWith(std::get<1>(it));
    }
    rewriter.eraseOp(outer_op);
  }
  return control_edges;
}

}  // namespace