%0 = "tf.Const"() {value = dense<true> : tensor<i1>} : () -> tensor<i1>
  func.return %0 : tensor<i1>
}

func.func @while_body(%arg0: tensor<*xf32>) -> (tensor<*xf32>) {
  %0 = "tf.SomeOp"(%arg0) : (tensor<*xf32>) -> tensor<*xf32>
  func.return %0 : tensor<*xf32>
}

// Test that -tf-drop-while-shape-invariant-in-device-cluster pass does not drop

// CHECK-SAME: dialect = "bar"
// CHECK: bar.a
// CHECK: bar.b
// CHECK: bar.c
// CHECK: func [[foo]]
// CHECK: foo.someop
// CHECK: call [[bar]]

// -----

func.func @handles_nested_regions(%arg0: f32, %arg1: f32) {
  %0 = "glue.someop"() : () -> f32
  "foo.someop"() ({
    "foo.some_other_op"() ({
      "bar.a"(%arg0) : (f32) -> f32
      "bar.b"(%arg1) : (f32) -> f32

  %graph:2 = tf_executor.graph {
    %value, %value_control = tf_executor.island wraps "tf.GetKey"() : () -> tensor<f32>
    %ret0, %ret0_control = tf_executor.island wraps "tf.SomeOp"() : () -> !tf_res
    %ret1, %ret1_control = tf_executor.island wraps "tf.SomeOp"() : () -> !tf_res
    %control_unknown = tf_executor.island wraps "tf.UnknownOp"() : () -> ()
    %key, %key_control = tf_executor.island wraps "tf.GetKey"() : () -> !tf_str

        %1 = "tf.opB"() : () -> tensor<f32>
        tf_executor.yield %1 : tensor<f32>
      }
// CHECK-NOT: PartitionedCall
// CHECK: "tf.opA"
// CHECK: "tf.opA"
// CHECK: "tf.SomeOp"

func.func private @handles_multiple_returns(%arg0: f32, %arg1: f32) -> (f32, f32) {
  "SomeOp"() : () -> ()
 ^bb0:
  return %arg0, %arg1 : f32, f32
 ^bb1:
  return %arg1, %arg0 : f32, f32
}

// -----

// CHECK-LABEL: @removes_multiple_returns
func.func private @removes_multiple_returns(%arg0: f32, %arg1: f32) -> (f32, f32) {
  "SomeOp"() : () -> ()
 ^bb0:
  // CHECK: return
  // CHECK-NOT: arg

      %3 = "tf.opA"(%arg0) : (tensor<i1>) -> tensor<i1>
      %4 = "tf.opA"(%3) : (tensor<i1>) -> tensor<i1>
      %5 = "tf.SomeOp"(%arg0, %1#0) : (tensor<i1>, tensor<f32>) -> tensor<i32>
      tf_executor.yield %4, %5 : tensor<i1>, tensor<i32>
    }
    tf_executor.fetch %2#0 : tensor<i1>
  }
  func.return %0 : tensor<i1>
}

// CHECK: module

// CHECK-LABEL: func @testControlOperands
func.func @testControlOperands() {
  %0:2 = tf_executor.graph {
    %1 = tf_executor.island {
      "tf.someOp"() : () -> ()
      tf_executor.yield
    }
    %2:3 = tf_executor.island(%1) {
      %3:2 = "tf_device.parallel_execute"() ({
        %4 = "tf.opA"() : () -> tensor<i1>
        tf_device.return %4 : tensor<i1>

// expected-error @+1 {{failed to legalize operation 'func.func' that was explicitly marked illegal}}
func.func @test_identity_wrong_type(%arg0: tensor<4x2x!tf_type.string>) -> tensor<4x2x!tf_type.stringref> {
  %0 = "tf.SomeOp"(%arg0) : (tensor<4x2x!tf_type.string>) -> tensor<4x2x!tf_type.stringref>
  func.return %0 : tensor<4x2x!tf_type.stringref>

func.func @simple_independent_chains_while_cond(%arg0: !tf_res, %arg1: !tf_res, %arg2: tensor<f32>, %arg3: tensor<f32>) -> (tensor<i32>) {
  %graph = tf_executor.graph {
    %island, %ctrl = tf_executor.island {
      %pred = "tf.SomeOp"(%arg2) : (tensor<f32>) -> tensor<i32>
      tf_executor.yield %pred : tensor<i32>
    }
    tf_executor.fetch %island : tensor<i32>
  }
  func.return %graph : tensor<i32>
}

  func.return
}

// CHECK-LABEL: func @resource
func.func @resource() {
  // CHECK: tf.SomeOp
  // CHECK-SAME: dtype = !corert.resource
  "tf.SomeOp"() {device = "/device:CPU:0", dtype = !tf_type.resource} : () -> ()
  func.return
}

// CHECK-LABEL: func @variant
func.func @variant(%arg: tensor<!tf_type.variant>) {