Searched refs:_generate_enqueue_op (Results 1 – 3 of 3) sorted by relevance
519 def _generate_enqueue_op(self, member in InfeedQueue625 self._generate_enqueue_op(743 self._generate_enqueue_op(
934 def _generate_enqueue_op( member in TPUEmbedding1278 enqueue_op = self._generate_enqueue_op(1310 enqueue_op = self._generate_enqueue_op(1326 enqueue_op = self._generate_enqueue_op(
1650 self._generate_enqueue_op( # pylint: disable=g-complex-comprehension1744 def _generate_enqueue_op(self, member in TPUEmbedding