Searched defs:embedding_tables (Results 1 – 4 of 4) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/python/tpu/ | ||
| H A D | tpu_embedding_base.py | 77 def embedding_tables(self): member in TPUEmbeddingBase |
| H A D | tpu_embedding_for_serving.py | 119 def embedding_tables( member in TPUEmbeddingForServing |
| H A D | tpu_embedding_v1.py | 143 def embedding_tables( member in TPUEmbeddingV0 |
| H A D | tpu_embedding_v2.py | 503 def embedding_tables( member in TPUEmbedding |