Searched defs:tpu_embedding (Results 1 – 1 of 1) sorted by relevance
51 def create_dummy_table_variables(tpu_embedding): argument101 def hook_dummy_table_variables_to_activations(tpu_embedding, activations, argument127 def get_gradients_through_dummy_table_variables(tpu_embedding): argument