%handle0, %handle0_control = tf_executor.island wraps "tf.SomeOp"() : () -> !tf_res
    %handle1, %handle1_control = tf_executor.island wraps "tf.SomeOp"() : () -> !tf_res
    %control_A = tf_executor.island wraps "tf.OpA"() : () -> ()
    %while_out:2, %while_control = tf_executor.island(%control_A) wraps "tf.While"(
            %handle0, %handle1) {
        body = @body, cond = @cond, is_stateless = false

	}

	if authZPluginCfg.URL == nil {
		opaCfg, err := opa.LookupConfig(s[config.PolicyOPASubSys][config.Default],
			NewHTTPTransport(), xhttp.DrainBody)
		if err != nil {
			iamLogIf(ctx, fmt.Errorf("Unable to initialize AuthZPlugin from legacy OPA config: %w", err))
		} else {
			authZPluginCfg.URL = opaCfg.URL
			authZPluginCfg.AuthToken = opaCfg.AuthToken

}

// CHECK-LABEL: func @variant_input_output_already_marked
func.func @variant_input_output_already_marked() {
  // CHECK: "tf.opA"
  // CHECK-SAME: _xla_outside_compilation = "0"
  // CHECK: "tf.opB"
  // CHECK-SAME: _xla_outside_compilation = "0"
  "tf_device.cluster"() ({
    %1= "tf.opA"() {_xla_outside_compilation = "0"} : () -> tensor<!tf_type.variant<tensor<f32>>>

replicas and is mapped to a single region argument.

Operands not replicated can be implicitly captured by ops in the region. Results
are replicated each from the regions terminator.

For example:
```
%0 = "tf.opA"() : () -> tensor<i32>
%1 = "tf.opB"() : () -> tensor<i32>
%2 = "tf.opC"() : () -> tensor<f32>
%3 = "tf.opD"() : () -> tensor<f32>
%4 = "tf.opE"() : () -> tensor<!tf_type.resource>

      tf_executor.yield
    }
    // CHECK: tf_executor.fetch
    tf_executor.fetch
  }
  func.return
}

func.func @then_function() {
  tf_executor.graph {
    tf_executor.island {
      "tf.OpA"() : () -> ()
      tf_executor.yield
    }
    tf_executor.fetch
  }
  func.return
}

func.func @else_function() {
  tf_executor.graph {
    tf_executor.island {
      "tf.OpB"() : () -> ()

    enableNamespacesByDefault: false
    # Mutations that occur after the sidecar injector are not handled by default, as the Istio sidecar injector is only run
    # once. For example, an OPA sidecar injected after the Istio sidecar will not have it's liveness/readiness probes rewritten.
    # Setting this to `IfNeeded` will result in the sidecar injector being run again if additional mutations occur.

  %0 = "tf.VarHandleOp"() {container = "c", shared_name = "v"} : () -> tensor<*x!tf_type.resource<tensor<2x8xi32>>>
  %1 = "tf.Const"() {value = dense<1> : tensor<i32>} : () -> tensor<i32>
  "tf_device.cluster"() ({
    "tf.OpA"(%0, %1) {f = @reachable_function} : (tensor<*x!tf_type.resource<tensor<2x8xi32>>>, tensor<i32>) -> ()
    tf_device.return
  }) : () -> ()
  func.return
}

// CHECK-LABEL: func @reachable_function

        // a role as connection between `tf.TPUReplicatedInput` or
        // `tf.TPUPartitionedInput`. For example, we have the follow pseudocode:
        // %0 = tf_executor.island wraps "tf.OpA" (){_replication_info = 'c'}
        // %1 = tf_executor.island wraps "tf.Identity(%0)
        // %2 = tf_executor.island wraps "tf.TPUReplicatedInput"(%1)

        if (!isa<TF::TPUReplicatedInputOp, TF::TPUPartitionedInputOp,

    enableNamespacesByDefault: false

    # Mutations that occur after the sidecar injector are not handled by default, as the Istio sidecar injector is only run
    # once. For example, an OPA sidecar injected after the Istio sidecar will not have it's liveness/readiness probes rewritten.
    # Setting this to `IfNeeded` will result in the sidecar injector being run again if additional mutations occur.

  }

  // CHECK-LABEL: func @xla_host_compute_shape_inferred
  func.func @xla_host_compute_shape_inferred() -> (tensor<1x1120x8xi32>, tensor<1x1120x8xi32>, tensor<1120x8xi32>, tensor<2xi32>) {
    %2335 = "tf.OpA"() : () -> tensor<3360x8xi32>
    %cst_2 = "tf.Const"() <{value = dense<[3, 1120, -1]> : tensor<3xi32>}> : () -> tensor<3xi32>
    %cst_68 = "tf.Const"() <{value = dense<0> : tensor<i32>}> : () -> tensor<i32>