Searched refs:embedding_tables (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/python/tpu/ |
| D | tpu_embedding_v2_cpu_test.py | 122 tables=mid_level.embedding_tables,
126 table = mid_level.embedding_tables[config.table].numpy()
140 tables=mid_level.embedding_tables,
164 tables=mid_level.embedding_tables,
168 table = mid_level.embedding_tables[config.table].numpy()
187 tables=mid_level.embedding_tables,
193 table = mid_level.embedding_tables[config.table].numpy()
216 tables=mid_level.embedding_tables,
238 tables=mid_level.embedding_tables,
244 table = mid_level.embedding_tables[config.table].numpy()
[all …]
|
| D | tpu_embedding_v2_test.py | 280 features, None, self.cpu_mid_level.embedding_tables,
537 set(mid_level_api.embedding_tables.keys()),
544 mid_level_api.embedding_tables()
|
| D | tpu_embedding_v2.py | 412 def embedding_tables( member in TPUEmbedding
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.tpu.experimental.embedding.-t-p-u-embedding.pbtxt | 8 name: "embedding_tables"
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.tpu.experimental.embedding.-t-p-u-embedding.pbtxt | 8 name: "embedding_tables"
|