// CHECK: RemoteCall
  // CHECK-SAME: (tensor<!tf_type.string>, tensor<!tf_type.string>, tensor<i64>) -> tensor<!tf_type.string>
  %0 = "tf.RemoteCall"(%arg1, %arg0, %arg2) <{f = @__inference__next_func_3760}> {device = ""} : (tensor<!tf_type.string>, tensor<!tf_type.string>, tensor<i64>) -> tensor<!tf_type.string>
  return
}

      %0 = "tf.opA"(%arg0) {_xla_compile_device_type = "TPU", _replication_info = "cluster"} : (tensor<i1>) -> tensor<i32>
      %1 = "tf.While"(%0) {body = @while_body_with_cluster_attr, cond = @while_cond_with_cluster_attr, is_stateless = false, name = "A", parallel_iterations = 10 : i64} : (tensor<i32>) -> tensor<i32>

    %replicated_out2, %replicated_control2 = tf_executor.island wraps "tf.TPUReplicatedInput"(%replicated_out1) {N = 1 : i64, T = i32, device = "", index = 0 : i64, is_mirrored_variable = false} : (tensor<4x4xf32>) -> tensor<4x4xf32>

"builtin.module"() ({
  "func.func"() ({
  ^bb0(%arg0: tensor<i32>):
    %1617 = "tf.While"(%arg0) {_lower_using_switch_merge = true, _num_original_outputs = 7 : i64, _read_only_resource_inputs = [], body = @main_while_body_4225150, cond = @main_while_cond_4225140, device = "", is_stateless = false, parallel_iterations = 10 : i64, shape_invariant} : (tensor<i32>) -> (tensor<i32>)
    "func.return"() : () -> ()

    %0 = "tf.Const"() {value = dense<1> : tensor<i64>} : () -> tensor<i64>
    %1 = "tf.Const"() {value = dense<0> : tensor<1x1xi32>} : () -> tensor<1x1xi32>

    %0 = stablehlo.convolution(%arg0, %arg1) dim_numbers = [b, 0, 1, f]x[0, 1, i, o]->[b, 0, 1, f], window = {pad = [[0, 1], [1, 1]]} {batch_group_count = 1 : i64, feature_group_count = 1 : i64} : (tensor<1x3x4x3xf32>, tensor<2x3x3x2xf32>) -> tensor<1x3x4x2xf32>
    return %0 : tensor<1x3x4x2xf32>
  }
}

// CHECK-LABEL: quantize_conv_per_tensor
// CHECK-SAME: %[[ARG0:.+]]: tensor<1x3x4x3xf32>

  }

  // CHECK-LABEL: multiple_return_values
  func.func @multiple_return_values(%arg0: tensor<3xi64>) -> tensor<i64> {
     %0:3 = "tf.Unpack"(%arg0) {axis = 0 : i64} : (tensor<3xi64>) -> (tensor<i64>, tensor<i64>, tensor<i64>)
    // CHECK: return %1 : tensor<i64>
    func.return %0#0 : tensor<i64>
  }

  // CHECK-LABEL: constant_parameter
  func.func @constant_parameter(%arg0: tensor<2xf32>) -> tensor<2xf32> {

  func.return %0 : tensor<1x1x5x6x7xf32>
  // CHECK-LABEL: strided_slice_big_dims

        %3 = "tf.While"(%1) {body = @while_body_without_cluster_attr, cond = @while_cond_with_cluster_attr, is_stateless = false, name = "C", parallel_iterations = 10 : i64} : (tensor<i32>) -> tensor<i32>

  /// TaggedValue constructor for type INT64.
  explicit TaggedValue(int64_t i64) : type_(INT64), data_(Int64(i64)) {}
  /// TaggedValue constructor for type FLOAT32.
  explicit TaggedValue(Float32 f32) : type_(FLOAT32), data_(f32) {}
  /// TaggedValue constructor for type INT64.
  explicit TaggedValue(Int64 i64) : type_(INT64), data_(i64) {}
  /// TaggedValue constructor for type STRING.