Resolver¶

AutoDist device-setting kernels.

class DeviceResolver(cluster)[source]¶

Bases: object

Resolving AutoDist DeviceSpec to TensorFlow DeviceSpec given a cluster.

resolve_to_device_spec(device)[source]¶: Resolve an AutoDist DeviceSpec or its string to a TensorFlow DeviceSpec.

resolve_to_device_str(device)[source]¶: Resolve an AutoDist DeviceSpec or its string to a TensorFlow device string.