Home
last modified time | relevance | path

Searched refs:embedding_var_name (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/python/tpu/
Dfeature_column.py400 def _record_variable_scope_and_name(embedding_var_name, argument
414 if embedding_var_name in var_def_dict:
415 if (var_def_dict[embedding_var_name][0] != captured_scope_name
419 'got {}; expected {}'.format(embedding_var_name,
421 var_def_dict[embedding_var_name][0]))
422 if var_def_dict[embedding_var_name][1] != embedding_var_name_in_fc:
425 'got {}; expected {}'.format(embedding_var_name,
427 var_def_dict[embedding_var_name][1]))
429 var_def_dict[embedding_var_name] = (captured_scope_name,
D_tpu_estimator_embedding.py48 def _get_table_name_from_embedding_var_name(embedding_var_name): argument
49 return '{}{}'.format(_TABLE_NAME_PREFIX, embedding_var_name)
93 embedding_var_name = _get_embedding_var_name_from_table_name(table_name)
94 (scope_name, var_name) = collection[0][embedding_var_name]