Searched refs:embedding_config (Results 1 – 10 of 10) sorted by relevance
306 def embedding_config(self): member in _InternalTPUContext310 embedding_config = self._lazy_embedding_config_dict[master]312 embedding_config = None314 embedding_config = _tpu_estimator_embedding.EmbeddingConfig(317 if not embedding_config.has_embedding_tables():318 embedding_config = None319 self._lazy_embedding_config_dict[master] = embedding_config321 if embedding_config is not None:326 embedding_config.tpu_embedding = embedding_config.get_tpu_embedding(mode)327 return embedding_config
454 if ctx.embedding_config is None:458 ctx.embedding_config.tpu_embedding.config_proto)537 embedding_config=self._embedding_layer_config))930 if ctx.embedding_config:932 ctx.embedding_config.tpu_embedding.generate_enqueue_ops(1416 if self._ctx.embedding_config:1417 tpu_embedding_ = self._ctx.embedding_config.tpu_embedding1423 self._ctx.embedding_config.dummy_table_variables))1473 if self._ctx.embedding_config is None:1476 tpu_embedding_ = self._ctx.embedding_config.tpu_embedding[all …]
279 if ctx.embedding_config:280 tpu_embedding_ = ctx.embedding_config.tpu_embedding
84 def initialize_system(embedding_config=None, job=None): argument99 config_string = ("" if embedding_config is None else100 embedding_config.SerializeToString())102 return tpu_ops.configure_distributed_tpu(embedding_config=config_string)
12 name: "embedding_config"
637 …argspec: "args=[\'embedding_config\', \'tpu_embedding_config\', \'is_global_init\', \'name\'], var…
6090 name: "embedding_config"
13854 name: "embedding_config"
13796 name: "embedding_config"