MCPcopy Create free account
hub / github.com/alibaba/nann / while_loop

Function while_loop

tensorflow/tensorflow/python/tpu/training_loop.py:30–178  ·  view source on GitHub ↗

Builds a training loop for TPUs. The set of loop-carried tensors corresponds to `inputs`. Both `condition` and `body` take the current value of the loop-carried tensors. 'body' additionally takes a tuple of infeed from infeed_queue if infeed_queue is not None. `condition` must return a s

(condition, body, inputs=None, infeed_queue=None, name=None)

Source from the content-addressed store, hash-verified

source not stored for this graph (policy: none)

Callers 1

repeatFunction · 0.70

Calls 3

TypeErrorFunction · 0.85
constantMethod · 0.45
while_loopMethod · 0.45

Tested by

no test coverage detected