Searched defs:_generate_enqueue_op (Results 1 – 3 of 3) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/python/tpu/ | ||
| H A D | tpu_feed.py | 519 def _generate_enqueue_op(self, member in InfeedQueue |
| H A D | tpu_embedding_v2.py | 936 def _generate_enqueue_op( member in TPUEmbedding |
| H A D | tpu_embedding.py | 1841 def _generate_enqueue_op(self, member in TPUEmbedding |