hub / github.com/alibaba/nann / _TPUInferenceContext

Class _TPUInferenceContext

tensorflow/tensorflow/python/tpu/tpu.py:1546–1581 · view source on GitHub ↗

A `ControlFlowContext` for nodes inside a TPU inference computation. The primary role of `TPUReplicateContext` is to sanity check operators inside a tpu.rewrite_for_inference() computation.

Source from the content-addressed store, hash-verified

source not stored for this graph (policy: none)

Callers 1

wrapped_computationFunction · 0.85

Calls

no outgoing calls

Tested by

no test coverage detected