Searched refs:InfeedQueue (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/python/tpu/ |
| D | tpu_infeed_test.py | 33 i = tpu_feed.InfeedQueue(number_of_tuple_elements=2)
38 i = tpu_feed.InfeedQueue(
45 i = tpu_feed.InfeedQueue(tuple_shapes=[[1], [2, 3]])
50 i = tpu_feed.InfeedQueue(shard_dimensions=[1, 0, 7])
57 i = tpu_feed.InfeedQueue()
59 i = tpu_feed.InfeedQueue(
62 i = tpu_feed.InfeedQueue(number_of_tuple_elements=2, tuple_shapes=[[1]])
64 i = tpu_feed.InfeedQueue(number_of_tuple_elements=2, shard_dimensions=[1])
66 i = tpu_feed.InfeedQueue(tuple_shapes=[[1], [2, 3]], shard_dimensions=[1])
70 i = tpu_feed.InfeedQueue(number_of_tuple_elements=2)
[all …]
|
| D | training_loop.py | 37 infeed_queue: Optional[tpu_feed.InfeedQueue] = None, argument
193 infeed_queue: Optional[tpu_feed.InfeedQueue] = None, argument
|
| D | tpu.py | 884 infeed_queue: Optional[tpu_feed.InfeedQueue] = None, argument
1177 infeed_queue: Optional[tpu_feed.InfeedQueue] = None, argument
1716 infeed_queue: Optional[tpu_feed.InfeedQueue] = None, argument
1873 infeed_queue: Optional[tpu_feed.InfeedQueue] = None, argument
1956 infeed_queue: Optional[tpu_feed.InfeedQueue] = None, argument
2017 infeed_queue: Optional[tpu_feed.InfeedQueue] = None, argument
2174 infeed_queue: Optional[tpu_feed.InfeedQueue] = None, argument
|
| D | tpu_feed.py | 125 class InfeedQueue(object): class
753 class _PartitionedInfeedQueue(InfeedQueue):
|
| D | tpu_test.py | 104 infeed = tpu_feed.InfeedQueue(
|
| /external/tensorflow/tensorflow/python/compiler/xla/ |
| D | xla_test.py | 278 queue = tpu_feed.InfeedQueue(2)
295 queue = tpu_feed.InfeedQueue(1)
314 queue = tpu_feed.InfeedQueue(1)
333 queue = tpu_feed.InfeedQueue(1)
|