OpenGrok
Home
Sort by relevance
Sort by last modified time
Full Search
Definition
Symbol
File Path
History
|
|
Help
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