%3 = "tf.StringToHashBucketFast"(%arg0) {device = "", num_buckets = 5 : i64} : (tensor<?x!tf_type.string>) -> tensor<?xi64>
    %4 = "tf.AddV2"(%3, %1) {device = ""} : (tensor<?xi64>, tensor<i64>) -> tensor<?xi64>
    %5 = "tf.LookupTableFindV2"(%0, %arg0, %cst) {device = ""} : (tensor<!tf_type.resource>, tensor<?x!tf_type.string>, tensor<i64>) -> tensor<*xi64>
    return %5 : tensor<*xi64>
  }

// Check that HashTable op is lifted.

}

func.func private @body(%arg: tensor<i32>, %handle: tensor<!tf_type.resource>) -> (tensor<i32>, tensor<!tf_type.resource>) {
  %default = "tf.Const"() {value = dense<16> : tensor<i32> } : () -> tensor<i32>
  %0 = "tf.LookupTableFindV2"(%handle, %arg, %default) : (tensor<!tf_type.resource>, tensor<i32>, tensor<i32>) -> tensor<i32>
  func.return %0, %handle : tensor<i32>, tensor<!tf_type.resource>
}

// CHECK-LABEL: func @set_readonly_stateless

      %out_1, %ctl_2 = tf_executor.island wraps "tf.LookupTableFindV2"(%out_0, %arg0, %out) {device = ""} : (tensor<!tf_type.resource>, tensor<?x!tf_type.string>, tensor<i64>) -> tensor<*xi64>
      tf_executor.fetch %out_1 : tensor<*xi64>
    }
    return %0 : tensor<*xi64>
  }

  // CHECK: [[v:%.*]]:2 = "tf._TfrtGetResource"() <{container = ["", ""], indices = [0, 1], shared_name = [{{.*}}, {{.*}}]}> {device = "/job:localhost/replica:0/task:0/device:CPU:0"}
  // CHECK: [[r:%.*]] = "tf.LookupTableFindV2"([[v]]#[[handle_idx]]
  // CHECK: return [[v]]#[[size_idx]], [[r]]

  "tf.LookupTableImportV2"(%handle, %keys, %values) {device = ""} : (tensor<!tf_type.resource>, tensor<4x!tf_type.string>, tensor<4xi64>) -> ()
  %value0 = "tf.LookupTableFindV2"(%handle, %key, %default) {device = "/job:localhost/replica:0/task:0/device:CPU:0"} : (tensor<!tf_type.resource>, tensor<!tf_type.string>, tensor<i64>) -> tensor<i64>

    // expected-remark@+1 {{Cost: 1024}}
    %1 = "tf.LookupTableFindV2"(%0, %key, %default) {device = "/job:localhost/replica:0/task:0/device:CPU:0"} : (tensor<!tf_type.resource>, tensor<?x!tf_type.string>, tensor<i64>) -> tensor<*xi64>
    // 17 = 1 + 8 + 8
    // expected-remark@+1 {{Cost: 17}}

  %default = "tf.Const"() {device = "/CPU:0", value = dense<0> : tensor<1xi32>} : () -> tensor<1xi32>
  // CHECK: tf.HashTableV2
  %0 = "tf.LookupTableFindV2"(%arg1, %arg0, %default) {device = "/device:CPU:0"} : (tensor<*x!tf_type.resource>, tensor<1xi32>, tensor<1xi32>) -> tensor<1xi32>
  func.return %0 : tensor<1xi32>
}

// CHECK-LABEL: func @main