Searched defs:tpu_embedding (Results 1 – 1 of 1) sorted by relevance
49 def create_dummy_table_variables(tpu_embedding): argument99 def hook_dummy_table_variables_to_activations(tpu_embedding, activations, argument125 def get_gradients_through_dummy_table_variables(tpu_embedding): argument