Home
last modified time | relevance | path

Searched defs:InfeedQueue (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/python/tpu/
H A Dtpu.py918 infeed_queue: Optional[tpu_feed.InfeedQueue] = None,
1210 infeed_queue: Optional[tpu_feed.InfeedQueue] = None,
1787 infeed_queue: Optional[tpu_feed.InfeedQueue] = None,
1951 infeed_queue: Optional[tpu_feed.InfeedQueue] = None,
2035 infeed_queue: Optional[tpu_feed.InfeedQueue] = None,
2097 infeed_queue: Optional[tpu_feed.InfeedQueue] = None,
2254 infeed_queue: Optional[tpu_feed.InfeedQueue] = None,
H A Dtraining_loop.py33 infeed_queue: Optional[tpu_feed.InfeedQueue] = None,
185 infeed_queue: Optional[tpu_feed.InfeedQueue] = None,
H A Dtpu_feed.py120 class InfeedQueue(object): class