MCPcopy Create free account
hub / github.com/tensorflow/lingvo / BuildTpuSubgraph

Method BuildTpuSubgraph

lingvo/core/program.py:327–337  ·  view source on GitHub ↗

Subclasses should construct a model/graph to be executed by Run. Specific to TPU execution, this may involve a @tpu_function.on_device_training_loop etc. Args: strategy: Intended for use by host-driven TPUStrategy programs, otherwise None.

(self, strategy=None)

Source from the content-addressed store, hash-verified

source not stored for this graph (policy: none)

Callers 2

__init__Method · 0.45
__init__Method · 0.45

Calls

no outgoing calls

Tested by

no test coverage detected