generate_enqueue_ops (reference) in projects: aosp_15_r20

Searched refs:generate_enqueue_ops (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/python/tpu/
H A D	tpu_feed.py	`572 def generate_enqueue_ops(self, member in InfeedQueue 818 def generate_enqueue_ops(self, sharded_inputs): member in _PartitionedInfeedQueue`
H A D	tpu_embedding_v2.py	`1262 def generate_enqueue_ops(): function 1286 tpu.outside_compilation(generate_enqueue_ops)`
H A D	tpu_embedding.py	`1722 def generate_enqueue_ops( member in TPUEmbedding`