Searched refs:initialize_system_for_tpu_embedding (Results 1 – 3 of 3) sorted by relevance
172 `initialize_system_for_tpu_embedding`.198 after `initialize_system_for_tpu_embedding` and before any data is enqueued
395 tpu.initialize_system_for_tpu_embedding(self._config_proto)
169 def initialize_system_for_tpu_embedding( function