message: key() returns a dict_key object in python 3 which doesn't have index() so switch to use list() instead, which extracts the list of keys in both python 2 and 3.
diff: @@ -119,7 +119,7 @@ def hook_dummy_table_variables_to_activations(tpu_embedding, activations,
  new_activations[feature] = tpu_ops.tpu_embedding_activations(
  dummy_table_variables[table],
  activations[feature],
  - table_id=tpu_embedding.table_to_config_dict.keys().index(table),
  + table_id=list(tpu_embedding.table_to_config_dict).index(table),
  lookup_id=tpu_embedding.table_to_features_dict[table].index(feature))
  return new_activations
  