Searched defs:while_loop (Results 1 - 4 of 4) sorted by relevance
/external/tensorflow/tensorflow/contrib/tpu/python/tpu/ |
H A D | training_loop.py | 29 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 D | standard_ops.py | 53 from tensorflow.python.ops.control_flow_ops import while_loop namespace
|
H A D | control_flow_ops.py | 28 @@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 D | while_test.cc | 942 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