Searched refs:infeed_queue (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/python/tpu/ |
| D | training_loop.py | 37 infeed_queue: Optional[tpu_feed.InfeedQueue] = None,
72 body, input_arity, infeed_queue)
74 if infeed_queue is None:
84 [i.name for i in inputs]), infeed_queue.number_of_tuple_elements,
89 if infeed_queue is None:
118 if infeed_queue:
125 infeed_queue.set_number_of_shards(number_of_shards)
126 dequeue_ops = [d for d in infeed_queue.generate_dequeue_op()]
193 infeed_queue: Optional[tpu_feed.InfeedQueue] = None,
230 cond, body_wrapper, inputs=inputs, infeed_queue=infeed_queue, name=name)
|
| D | tpu.py | 884 infeed_queue: Optional[tpu_feed.InfeedQueue] = None,
980 infeed_queue,
1177 infeed_queue: Optional[tpu_feed.InfeedQueue] = None,
1316 computation, input_arity, infeed_queue)
1318 if infeed_queue is None:
1329 for i in inputs[0]]), infeed_queue.number_of_tuple_elements,
1334 if infeed_queue:
1450 if infeed_queue is not None:
1451 infeed_queue.set_number_of_shards(num_replicas)
1452 for t in infeed_queue.generate_dequeue_op():
[all …]
|
| D | tpu_test.py | 109 return training_loop.repeat(5, training_step, infeed_queue=infeed)
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.tpu.pbtxt | 21 …argspec: "args=[\'computation\', \'inputs\', \'num_shards\', \'infeed_queue\', \'device_assignment…
45 …argspec: "args=[\'computation\', \'inputs\', \'infeed_queue\', \'device_assignment\', \'name\', \'…
49 …argspec: "args=[\'computation\', \'inputs\', \'infeed_queue\', \'device_assignment\', \'name\', \'…
53 …t_shard_axes\', \'outputs_from_all_shards\', \'output_shard_axes\', \'infeed_queue\', \'device_ass…
|
| /external/tensorflow/tensorflow/python/compiler/xla/ |
| D | xla.py | 596 def check_function_argument_count(func, input_arity, infeed_queue): argument
615 if infeed_queue is not None:
616 num_args_supplied += infeed_queue.number_of_tuple_elements
|