EagerSession.DevicePlacementPolicy

public static final enum EagerSession.DevicePlacementPolicy

Controls how to act when we try to run an operation on a given device but some input tensors are not on that device.

Inherited Methods

Enum Values

public static final EagerSession.DevicePlacementPolicy EXPLICIT

Running operations with input tensors on the wrong device will fail.

public static final EagerSession.DevicePlacementPolicy SILENT

Silently copy the tensor, which has a performance cost since the operation will be blocked till the copy completes. This is the default placement policy.

public static final EagerSession.DevicePlacementPolicy SILENT_FOR_INT32

Placement policy which silently copies int32 tensors but not other dtypes.

public static final EagerSession.DevicePlacementPolicy WARN

Copy the tensor to the right device but log a warning.