Searched refs:initialize_system_for_tpu_embedding (Results 1 – 3 of 3) sorted by relevance
239 saved_fn = tpu.initialize_system_for_tpu_embedding240 tpu.initialize_system_for_tpu_embedding = lambda x: None247 tpu.initialize_system_for_tpu_embedding = saved_fn
385 tpu.initialize_system_for_tpu_embedding(self._config_proto)
153 def initialize_system_for_tpu_embedding( function