hub / github.com/tensorflow/lingvo / WhileLoop

Function WhileLoop

lingvo/core/py_utils.py:6103–6142 · view source on GitHub ↗

Helper to construct a while loop. Args: cond: A callable NestedMap -> tf.bool. body: A callable NestedMap -> NestedMap. loop_state: A flattenable (NestedMap, list, tuple, etc.) representing the loop state. Returns: The final loop state in the same structure as loop_state.

(cond, body, loop_state)

Source from the content-addressed store, hash-verified

source not stored for this graph (policy: none)

Callers 1

ForLoopFunction · 0.85

Calls 6

TensorSpecsFunction · 0.85

FunctionClass · 0.85

FlattenFunction · 0.85

IsEagerModeFunction · 0.85

PackFunction · 0.85

AddFrameworkInputsMethod · 0.80

Tested by

no test coverage detected

Used in the wild real call sites across dependent graphs

searching dependent graphs…