// CHECK: [[VAL:%.*]] = "tf.HashTableV2"()
  // CHECK: "tf.LookupTableImportV2"([[VAL]], [[CST]], [[CST_0]])
}

// Tests that the tf.InitializeTableFromTextFileV2 op with explicit vocab size.

func.func @init_all_tables_with_explicit_vocab_size() {
  %cst = arith.constant dense<"tokens.txt"> : tensor<!tf_type.string>

    // CHECK: [[HASHTABLE:%.+]] = "tf.HashTableV2"()
    // CHECK: "tf.InitializeTableFromTextFileV2"([[HASHTABLE]], [[CST]])
    // CHECK: [[HASHTABLE_1:%.+]] = "tf.HashTableV2"()
    // CHECK: "tf.InitializeTableFromTextFileV2"([[HASHTABLE_1]], [[CST_1]])
    func.return
  }
}

// -----

// CHECK:  operator_codes: [ {
// CHECK:    deprecated_builtin_code: 32,
// CHECK:    custom_code: "HashTableV2",
// CHECK:    builtin_code: CUSTOM
// CHECK: } ],
// CHECK: subgraphs: [ {
// CHECK:   tensors: [ {
// CHECK:     shape: [  ],
// CHECK:     type: RESOURCE,
// CHECK:     buffer: 1,
// CHECK:     name: "tf.HashTableV2",
// CHECK:     quantization: {
// CHECK-EMPTY
// CHECK:     }
// CHECK:   } ],

module attributes {tf_saved_model.semantics} {

// Test hoisting hash table op.

// CHECK-LABEL: func @_tfrt_resource_init
// CHECK: [[handle:%.*]] = "tf.HashTableV2"()
// CHECK-SAME: shared_name = "x"
// CHECK: "tf._TfrtSetResource"([[handle]]) <{index = [[handle_id:.*]] : i64}> {device = "/job:localhost/replica:0/task:0/device:CPU:0"}
// CHECK: [[x:%.*]] = "tf.LookupTableSizeV2"([[handle]])

  func.return
  // CHECK: [[VAL:%.*]] = "tf.HashTableV2"()
  // CHECK: tf.InitializeTableFromTextFileV2"
}

// -----

// Tests that the tf.InitializeTableFromTextFileV2 op is not converted since
// unsupported value_index, 0.

func.func @init_all_tables() {
  %cst = arith.constant dense<"vocab_file_does_not_exist.txt"> : tensor<!tf_type.string>

    %cst_6 = "tf.Const"() {value = dense<0.00156862743> : tensor<f32>} : () -> tensor<f32>
    %0 = "tf.HashTableV2"() {container = "", device = "", key_dtype = !tf_type.string, shared_name = "hash_table_1", use_node_name_sharing = false, value_dtype = i64} : () -> tensor<!tf_type.resource>

  // CHECK: [[VAL:%.*]] = "tf.HashTableV2"()
  // CHECK: "tf.LookupTableImportV2"([[VAL]], [[CST]], [[CST_0]])
}

// Tests that the tf.InitializeTableFromTextFileV2 op with explicit vocab size.

func.func @init_all_tables_with_explicit_vocab_size() {
  %cst = arith.constant dense<"%FILE_PLACEHOLDER"> : tensor<!tf_type.string>

  // CHECK-NEXT:  "tf.HashTableV2"
  // CHECK-NEXT:  "tf.LookupTableRemoveV2"
  func.return
}

// -----

// Test for case with import op.
func.func @hashtable_import(%arg0: tensor<5x!tf_type.string>) {
  %cst = arith.constant dense<["emerson", "lake", "palmer"]> : tensor<3x!tf_type.string>
  %cst_0 = arith.constant dense<[0, 1, 2]> : tensor<3xi64>

    }
  }
}

# CHECK: tf.HashTableV2
# CHECK-SAME: shared_name = "hash_table_node"

# CHECK: "tf.Variable"
# CHECK-SAME: shared_name = "variable_node"
# CHECK: "tf.VariableV2"
# CHECK-SAME: shared_name = "variable_v2_node"

# CHECK: "tf.BatchFunction"
# CHECK-SAME: shared_name = "batch_node"

# CHECK: func private @create_resource
# CHECK: tf.HashTableV2

    %cst = arith.constant dense<[1, 2, 3, 4]> : tensor<4xi64>
    %cst_0 = arith.constant dense<["a", "b", "c", "d"]> : tensor<4x!tf_type.string>
    %0 = "tf.HashTableV2"() {container = "", device = "", key_dtype = i64, shared_name = "hash_table_dba2ccaa-f1b1-46d6-b276-98008f69da71", use_node_name_sharing = false, value_dtype = !tf_type.string} : () -> tensor<!tf_type.resource>