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

Method FPropTower

lingvo/core/base_model.py:544–562  ·  view source on GitHub ↗

Forward propagation through one tower of the model. Args: theta: A `.NestedMap` object containing variable values of this task copied to this tower's devices. input_batch: A `.NestedMap` object containing input tensors to this tower. Returns: (dict, dict):

(self, theta, input_batch)

Source from the content-addressed store, hash-verified

source not stored for this graph (policy: none)

Callers 2

_FPropTpuMethod · 0.95
_FPropSplitInputBatchMethod · 0.95

Calls 2

ComputePredictionsMethod · 0.95
ComputeLossMethod · 0.95

Tested by

no test coverage detected