%1 = "tf.A"() : () -> (tensor<2x!tf_type.string>)
     "tf_device.launch"() ({
       "tf.TPUExecuteAndUpdateVariables"(%arg1, %1) {device_var_reads_indices = [0], device_var_updates_indices = [-1]} : (tensor<*x!tf_type.resource<tensor<4xf32>>>, tensor<2x!tf_type.string>) -> ()
       tf_device.return
    }) {device = "TPU_REPLICATED_CORE_0"} : () -> ()

    "tf.TPUExecuteAndUpdateVariables"(%4, %arg0) {
        device = "foobar",
        device_var_reads_indices = [],
        device_var_updates_indices = []} : (
                tensor<2048xi32>,
                tensor<2x!tf_type.string>) -> ()
    return
}

// -----

// CHECK-LABEL test_identity

    %read2, %read2_control  = tf_executor.island wraps "tf.ReadVariableOp"(%arg1) : (!tf_res) -> tensor<f32>
    tf_executor.island wraps "tf.TPUExecuteAndUpdateVariables"(%ret0, %ret1, %key) {
        device_var_reads_indices = [0, 1],
        device_var_updates_indices = [0, 1]} : (!tf_res, !tf_res, !tf_str) -> ()
    tf_executor.fetch %ret0, %ret1: !tf_res, !tf_res
  }
  func.return %graph#0, %graph#1 : !tf_res, !tf_res
}