}

// -----

func.func @parallel_execute_regions_with_data_results(%arg0: tensor<i32>) -> tensor<i32> {
  // expected-error @+1 {{'func' attribute refers to an undefined function: undefined_func}}
  %0 = "tf_device.cluster_func"(%arg0) {func = @undefined_func} : (tensor<i32>) -> tensor<i32>
  func.return %0 : tensor<i32>

  let summary = "Invokes a program via IFRT on a device";

  let description = [{
    This op calls an IFRT program uniquely identified by the given program id.

    During lowering from a `tf_device.cluster_func` op to a `tf.IfrtCall` op,
    the region owned by the former will be outlined to a function with a
    `tfrt_ifrt_serving.program_id` attribute. After that, the runtime ensures

    This pass outlines the body of a `tf_device.cluster` into a function and
    replaces the `tf_device.cluster` op with an equivalent `tf_device.cluster_func`
    op. Implicit operands will be captured and materialized as explicit arguments to
    the newly created functions and associated `tf_device.cluster_func` ops.

    For example, the following:

    ```mlir
    func @computation(%arg0: tensor<i32>) -> tensor<i32> {

    %5 = "tf.ReadVariableOp"(%arg7) : (tensor<!tf_type.resource<tensor<f32>>>) -> tensor<f32>
    %6 = "tf.ReadVariableOp"(%arg9) : (tensor<!tf_type.resource<tensor<i64>>>) -> tensor<i64>

#include "tensorflow/compiler/mlir/tensorflow/ir/tf_device.cc.inc"
      >();

  addInterfaces<TFInlinerInterface>();
}

//===----------------------------------------------------------------------===//
// tf_device.cluster_func
//===----------------------------------------------------------------------===//

LogicalResult ClusterFuncOp::verifySymbolUses(
    mlir::SymbolTableCollection& symbolTable) {

    %1 = "tf.Shape"(%arg1) {_tpu_replicate = "cluster_Fn", _xla_outside_compilation = "0", device = ""} : (tensor<*x!tf_type.string>) -> tensor<*xi32>
    %2:3 = "tf.UnpackHyp"(%arg1) {_tpu_replicate = "cluster_Fn", _xla_outside_compilation = "0", device = "", max_seq_length = 16 : i64} : (tensor<*x!tf_type.string>) -> (tensor<*xi32>, tensor<*xi32>, tensor<*xf32>)

                            Operation* parent, MetadataMap& metadata_map) {
  std::string cluster_succ = "";
  if (op->hasAttr(TF::kTpuReplicateAttr)) {
    cluster_succ = op->getAttrOfType<StringAttr>(TF::kTpuReplicateAttr).str();
  }
  if (cluster_succ.empty()) {
    // TODO (b/269195256#comment16): Change to error after resolving issue