_generate_enqueue_op (reference) in projects: aosp_15_r20

Searched refs:_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 625 self._generate_enqueue_op( 744 self._generate_enqueue_op(`
H A D	tpu_embedding_v2.py	`936 def _generate_enqueue_op( member in TPUEmbedding 1274 enqueue_op = self._generate_enqueue_op( 1307 enqueue_op = self._generate_enqueue_op( 1324 enqueue_op = self._generate_enqueue_op(`
H A D	tpu_embedding.py	`1747 self._generate_enqueue_op( # pylint: disable=g-complex-comprehension 1841 def _generate_enqueue_op(self, member in TPUEmbedding`