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

Method _FPropSplitInputBatch

lingvo/core/base_model.py:610–649  ·  view source on GitHub ↗

Splits the input batch on the input device.

(self, theta, input_batch)

Source from the content-addressed store, hash-verified

source not stored for this graph (policy: none)

Callers 1

FPropMethod · 0.95

Calls 3

_FPropTpuMethod · 0.95
FPropTowerMethod · 0.95
WorkerDeviceInModelSplitMethod · 0.80

Tested by

no test coverage detected