hub / github.com/tensorflow/lingvo / reset

Method reset

lingvo/core/tpu_embedding_manager.py:71–92 · view source on GitHub ↗

Resets object state. In particular, removes the reference to the TPUEmbedding object, if present, necessary for re-running unit tests with a fresh API object.

(self)

Source from the content-addressed store, hash-verified

source not stored for this graph (policy: none)

Callers 6

__init__Method · 0.95

_LoopMethod · 0.80

testLayerStackMethod · 0.80

testLayerStackSummaryMethod · 0.80

testParallelDecSelfAttentionRelativeBiasFFNMethod · 0.80

reset_tpu_clusterMethod · 0.80

Calls

no outgoing calls

Tested by 3

testLayerStackMethod · 0.64

testLayerStackSummaryMethod · 0.64

testParallelDecSelfAttentionRelativeBiasFFNMethod · 0.64