HomeSort by relevance Sort by last modified time
    Searched refs:infeed_queue (Results 1 - 6 of 6) sorted by null

  /external/tensorflow/tensorflow/python/tpu/
xla.py 61 def check_function_argument_count(func, input_arity, infeed_queue):
68 infeed_queue: if not None, the infeed queue that will supply
80 if infeed_queue is not None:
81 num_args_supplied += infeed_queue.number_of_tuple_elements
training_loop.py 30 def while_loop(condition, body, inputs=None, infeed_queue=None, name=None):
36 infeed_queue if infeed_queue is not None. `condition` must return a
46 infeed_queue: if not None, the infeed queue from which to append a tuple
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
    [all...]
tpu.py 516 infeed_queue=None,
531 infeed_queue: If not `None`, the `InfeedQueue` from which to append a tuple
572 infeed_queue,
663 infeed_queue=None,
685 infeed_queue: If not `None`, the `InfeedQueue` from which to append a tuple
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:
    [all...]
tpu_test.py 81 return training_loop.repeat(5, training_step, infeed_queue=infeed)
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(
    [all...]
  /external/tensorflow/tensorflow/contrib/compiler/
xla.py 793 def check_function_argument_count(func, input_arity, infeed_queue):
800 infeed_queue: if not None, the infeed queue that will supply
812 if infeed_queue is not None:
813 num_args_supplied += infeed_queue.number_of_tuple_elements

Completed in 2015 milliseconds