Searched defs:tpu_embedding (Results 1 – 1 of 1) sorted by relevance
53 def create_dummy_table_variables(tpu_embedding): argument103 def hook_dummy_table_variables_to_activations(tpu_embedding, activations, argument129 def get_gradients_through_dummy_table_variables(tpu_embedding): argument