Home
last modified time | relevance | path

Searched refs:embedding_tables (Results 1 – 14 of 14) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/python/tpu/
H A Dtpu_embedding_for_serving_test.py117 table = mid_level.embedding_tables[config.table].numpy()
151 table = mid_level.embedding_tables[config.table].numpy()
172 table = mid_level.embedding_tables[config.table].numpy()
215 table = mid_level.embedding_tables[config.table].numpy()
268 mid_level.embedding_tables[self.table_user].numpy(),
289 mid_level.embedding_tables[self.table_user].numpy(),
H A Dtpu_embedding_for_serving.py119 def embedding_tables( member in TPUEmbeddingForServing
172 return cpu_embedding_lookup(features, weights, self.embedding_tables,
H A Dtpu_embedding_v1.py143 def embedding_tables( member in TPUEmbeddingV0
272 table = self.embedding_tables[feature.table]
H A Dtpu_embedding_base.py77 def embedding_tables(self): member in TPUEmbeddingBase
H A Dtpu_embedding_v2.py503 def embedding_tables( member in TPUEmbedding
/aosp_15_r20/external/tensorflow/tensorflow/python/tpu/tests/
H A Dtpu_embedding_v1_correctness_test.py148 tape.watch(mid_level_api.embedding_tables.values())
153 mid_level_api.embedding_tables.values())
155 list(zip(gradients, mid_level_api.embedding_tables.values())))
H A Dtpu_embedding_v1_checkpoint_test.py139 features, None, cpu_mid_level.embedding_tables,
H A Dtpu_embedding_v2_checkpoint_test.py296 features, None, cpu_mid_level.embedding_tables,
/aosp_15_r20/external/tensorflow/tensorflow/tools/api/golden/v2/
H A Dtensorflow.tpu.experimental.embedding.-t-p-u-embedding-for-serving.pbtxt9 name: "embedding_tables"
H A Dtensorflow.tpu.experimental.embedding.-t-p-u-embedding-v0.pbtxt9 name: "embedding_tables"
H A Dtensorflow.tpu.experimental.embedding.-t-p-u-embedding.pbtxt8 name: "embedding_tables"
/aosp_15_r20/external/tensorflow/tensorflow/tools/api/golden/v1/
H A Dtensorflow.tpu.experimental.embedding.-t-p-u-embedding-v0.pbtxt9 name: "embedding_tables"
H A Dtensorflow.tpu.experimental.embedding.-t-p-u-embedding-for-serving.pbtxt9 name: "embedding_tables"
H A Dtensorflow.tpu.experimental.embedding.-t-p-u-embedding.pbtxt8 name: "embedding_tables"