Searched refs:infeed_queue (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/python/tpu/ |
| D | training_loop.py | 30 def while_loop(condition, body, inputs=None, infeed_queue=None, name=None): argument
64 body, input_arity, infeed_queue)
66 if infeed_queue is None:
76 [i.name for i in inputs]), infeed_queue.number_of_tuple_elements,
81 if infeed_queue is None:
110 if infeed_queue:
117 infeed_queue.set_number_of_shards(number_of_shards)
118 dequeue_ops = [d for d in infeed_queue.generate_dequeue_op()]
181 def repeat(n, body, inputs=None, infeed_queue=None, name=None): argument
216 cond, body_wrapper, inputs=inputs, infeed_queue=infeed_queue, name=name)
|
| D | tpu.py | 516 infeed_queue=None, argument
572 infeed_queue,
663 infeed_queue=None, argument
779 computation, input_arity, infeed_queue)
781 if infeed_queue is None:
792 for i in inputs[0]]), infeed_queue.number_of_tuple_elements,
796 if infeed_queue:
868 if infeed_queue is not None:
869 infeed_queue.set_number_of_shards(num_replicas)
870 for t in infeed_queue.generate_dequeue_op():
[all …]
|
| D | xla.py | 61 def check_function_argument_count(func, input_arity, infeed_queue): argument
80 if infeed_queue is not None:
81 num_args_supplied += infeed_queue.number_of_tuple_elements
|
| D | tpu_estimator.py | 763 infeed_queue = tpu_feed.InfeedQueue(
765 captured_infeed_queue.capture(infeed_queue)
767 per_host_enqueue_ops = infeed_queue.generate_enqueue_ops(
827 infeed_queue = tpu_feed.InfeedQueue(
831 captured_infeed_queue.capture(infeed_queue)
832 infeed_queue.set_number_of_shards(num_of_replicas_per_host)
834 infeed_queue.split_inputs_and_generate_enqueue_ops(
915 infeed_queue = tpu_feed._PartitionedInfeedQueue(
920 per_host_enqueue_ops = infeed_queue.generate_enqueue_ops(
923 infeed_queue = tpu_feed.InfeedQueue(
[all …]
|
| D | tpu_test.py | 81 return training_loop.repeat(5, training_step, infeed_queue=infeed)
|
| /external/tensorflow/tensorflow/contrib/compiler/ |
| D | xla.py | 793 def check_function_argument_count(func, input_arity, infeed_queue): argument
812 if infeed_queue is not None:
813 num_args_supplied += infeed_queue.number_of_tuple_elements
|