Searched defs:while_loop (Results 1 - 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/contrib/tpu/python/tpu/
H A Dtraining_loop.py29 def while_loop(condition, body, inputs=None, infeed_queue=None, name=None): function
168 return control_flow_ops.while_loop(condition_wrapper, body_wrapper, inputs,
206 outputs = while_loop(
/external/tensorflow/tensorflow/python/ops/
H A Dstandard_ops.py53 from tensorflow.python.ops.control_flow_ops import while_loop namespace
H A Dcontrol_flow_ops.py28 @@while_loop
629 "argument of tf.while_loop or set_shape() on the loop variables." %
652 "`shape_invariants` argument of tf.while_loop or set_shape() "
670 "the `shape_invariants` argument of tf.while_loop or set_shape() "
714 """Calculate a max_size for use by stack ops inside an XLA while_loop.
717 value: The value inside the while_loop forward context. Used for printing
721 inside e.g. a cond context inside the while_loop.
727 ValueError: If `value` is nested inside a `while_loop` that either
731 - is inside a `while_loop` that is a parent of the calling context, and
749 "XLA while_loop becaus
3051 def while_loop(cond, function
[all...]
/external/tensorflow/tensorflow/compiler/xla/tests/
H A Dwhile_test.cc942 auto while_loop = [this, &body, build_condition](int count) { local
951 TF_ASSERT_OK_AND_ASSIGN(auto computation, while_loop(i));

Completed in 180 milliseconds