hub / github.com/tensorflow/lingvo / WorkerDeviceInModelSplit

Method WorkerDeviceInModelSplit

lingvo/core/cluster.py:493–515 · view source on GitHub ↗

Returns the device to use for 'device_index' for the current model split. Args: device_index: An int, the device index within 'model_split'. Returns: A string. The device to place ops onto. Raises: ValueError: if split_id of cluster is incorrectly set.

(self, device_index)

Source from the content-addressed store, hash-verified

source not stored for this graph (policy: none)

Callers 14

FPropMethod · 0.80

__init__Method · 0.80

FPropMethod · 0.80

_FPropSplitInputBatchMethod · 0.80

FPropMethod · 0.80

_GetDeviceOfLayerMethod · 0.80

CreatorFunction · 0.80

testWorkerDeviceInModelSplitSyncMethod · 0.80

testWorkerDeviceInModelSplitMethod · 0.80

ComputePredictionsMethod · 0.80

Calls

no outgoing calls

Tested by 2

testWorkerDeviceInModelSplitSyncMethod · 0.64

testWorkerDeviceInModelSplitMethod · 0.64