Searched refs:generate_enqueue_ops (Results 1 – 3 of 3) sorted by relevance
547 def generate_enqueue_ops(self, member in InfeedQueue792 def generate_enqueue_ops(self, per_host_sharded_inputs): member in _PartitionedInfeedQueue
767 per_host_enqueue_ops = infeed_queue.generate_enqueue_ops(920 per_host_enqueue_ops = infeed_queue.generate_enqueue_ops(925 per_host_enqueue_ops = infeed_queue.generate_enqueue_ops(932 ctx.embedding_config.tpu_embedding.generate_enqueue_ops(1021 enqueue_ops = infeed_queue.generate_enqueue_ops(
567 def generate_enqueue_ops(self, sparse_features_list): member in TPUEmbedding