GPUOptions.Experimental.Builder

public static final class GPUOptions.Experimental.Builder

Protobuf type tensorflow.GPUOptions.Experimental

Public Methods

GPUOptions.Experimental.Builder
addAllVirtualDevices (Iterable<? extends GPUOptions.Experimental.VirtualDevices > values)
 The multi virtual device settings.
GPUOptions.Experimental.Builder
addRepeatedField (com.google.protobuf.Descriptors.FieldDescriptor field, Object value)
GPUOptions.Experimental.Builder
addVirtualDevices ( GPUOptions.Experimental.VirtualDevices.Builder builderForValue)
 The multi virtual device settings.
GPUOptions.Experimental.Builder
addVirtualDevices (int index, GPUOptions.Experimental.VirtualDevices value)
 The multi virtual device settings.
GPUOptions.Experimental.Builder
addVirtualDevices ( GPUOptions.Experimental.VirtualDevices value)
 The multi virtual device settings.
GPUOptions.Experimental.Builder
addVirtualDevices (int index, GPUOptions.Experimental.VirtualDevices.Builder builderForValue)
 The multi virtual device settings.
GPUOptions.Experimental.VirtualDevices.Builder
addVirtualDevicesBuilder ()
 The multi virtual device settings.
GPUOptions.Experimental.VirtualDevices.Builder
addVirtualDevicesBuilder (int index)
 The multi virtual device settings.
GPUOptions.Experimental
GPUOptions.Experimental
GPUOptions.Experimental.Builder
GPUOptions.Experimental.Builder
clearCollectiveRingOrder ()
 If non-empty, defines a good GPU ring order on a single worker based on
 device interconnect.
GPUOptions.Experimental.Builder
clearField (com.google.protobuf.Descriptors.FieldDescriptor field)
GPUOptions.Experimental.Builder
clearKernelTrackerMaxBytes ()
 If kernel_tracker_max_bytes = n > 0, then a tracking event is
 inserted after every series of kernels allocating a sum of
 memory >= n.
GPUOptions.Experimental.Builder
clearKernelTrackerMaxInterval ()
 Parameters for GPUKernelTracker.
GPUOptions.Experimental.Builder
clearKernelTrackerMaxPending ()
 If kernel_tracker_max_pending > 0 then no more than this many
 tracking events can be outstanding at a time.
GPUOptions.Experimental.Builder
clearNumDevToDevCopyStreams ()
 If > 1, the number of device-to-device copy streams to create
 for each GPUDevice.
GPUOptions.Experimental.Builder
clearOneof (com.google.protobuf.Descriptors.OneofDescriptor oneof)
GPUOptions.Experimental.Builder
clearTimestampedAllocator ()
 If true then extra work is done by GPUDevice and GPUBFCAllocator to
 keep track of when GPU memory is freed and when kernels actually
 complete so that we can know when a nominally free memory chunk
 is really not subject to pending use.
GPUOptions.Experimental.Builder
clearUseUnifiedMemory ()
 If true, uses CUDA unified memory for memory allocations.
GPUOptions.Experimental.Builder
clearVirtualDevices ()
 The multi virtual device settings.
GPUOptions.Experimental.Builder
String
getCollectiveRingOrder ()
 If non-empty, defines a good GPU ring order on a single worker based on
 device interconnect.
com.google.protobuf.ByteString
getCollectiveRingOrderBytes ()
 If non-empty, defines a good GPU ring order on a single worker based on
 device interconnect.
GPUOptions.Experimental
final static com.google.protobuf.Descriptors.Descriptor
com.google.protobuf.Descriptors.Descriptor
int
getKernelTrackerMaxBytes ()
 If kernel_tracker_max_bytes = n > 0, then a tracking event is
 inserted after every series of kernels allocating a sum of
 memory >= n.
int
getKernelTrackerMaxInterval ()
 Parameters for GPUKernelTracker.
int
getKernelTrackerMaxPending ()
 If kernel_tracker_max_pending > 0 then no more than this many
 tracking events can be outstanding at a time.
int
getNumDevToDevCopyStreams ()
 If > 1, the number of device-to-device copy streams to create
 for each GPUDevice.
boolean
getTimestampedAllocator ()
 If true then extra work is done by GPUDevice and GPUBFCAllocator to
 keep track of when GPU memory is freed and when kernels actually
 complete so that we can know when a nominally free memory chunk
 is really not subject to pending use.
boolean
getUseUnifiedMemory ()
 If true, uses CUDA unified memory for memory allocations.
GPUOptions.Experimental.VirtualDevices
getVirtualDevices (int index)
 The multi virtual device settings.
GPUOptions.Experimental.VirtualDevices.Builder
getVirtualDevicesBuilder (int index)
 The multi virtual device settings.
List< GPUOptions.Experimental.VirtualDevices.Builder >
getVirtualDevicesBuilderList ()
 The multi virtual device settings.
int
getVirtualDevicesCount ()
 The multi virtual device settings.
List< GPUOptions.Experimental.VirtualDevices >
getVirtualDevicesList ()
 The multi virtual device settings.
GPUOptions.Experimental.VirtualDevicesOrBuilder
getVirtualDevicesOrBuilder (int index)
 The multi virtual device settings.
List<? extends GPUOptions.Experimental.VirtualDevicesOrBuilder >
getVirtualDevicesOrBuilderList ()
 The multi virtual device settings.
final boolean
GPUOptions.Experimental.Builder
mergeFrom (com.google.protobuf.Message other)
GPUOptions.Experimental.Builder
mergeFrom (com.google.protobuf.CodedInputStream input, com.google.protobuf.ExtensionRegistryLite extensionRegistry)
final GPUOptions.Experimental.Builder
mergeUnknownFields (com.google.protobuf.UnknownFieldSet unknownFields)
GPUOptions.Experimental.Builder
removeVirtualDevices (int index)
 The multi virtual device settings.
GPUOptions.Experimental.Builder
setCollectiveRingOrder (String value)
 If non-empty, defines a good GPU ring order on a single worker based on
 device interconnect.
GPUOptions.Experimental.Builder
setCollectiveRingOrderBytes (com.google.protobuf.ByteString value)
 If non-empty, defines a good GPU ring order on a single worker based on
 device interconnect.
GPUOptions.Experimental.Builder
setField (com.google.protobuf.Descriptors.FieldDescriptor field, Object value)
GPUOptions.Experimental.Builder
setKernelTrackerMaxBytes (int value)
 If kernel_tracker_max_bytes = n > 0, then a tracking event is
 inserted after every series of kernels allocating a sum of
 memory >= n.
GPUOptions.Experimental.Builder
setKernelTrackerMaxInterval (int value)
 Parameters for GPUKernelTracker.
GPUOptions.Experimental.Builder
setKernelTrackerMaxPending (int value)
 If kernel_tracker_max_pending > 0 then no more than this many
 tracking events can be outstanding at a time.
GPUOptions.Experimental.Builder
setNumDevToDevCopyStreams (int value)
 If > 1, the number of device-to-device copy streams to create
 for each GPUDevice.
GPUOptions.Experimental.Builder
setRepeatedField (com.google.protobuf.Descriptors.FieldDescriptor field, int index, Object value)
GPUOptions.Experimental.Builder
setTimestampedAllocator (boolean value)
 If true then extra work is done by GPUDevice and GPUBFCAllocator to
 keep track of when GPU memory is freed and when kernels actually
 complete so that we can know when a nominally free memory chunk
 is really not subject to pending use.
final GPUOptions.Experimental.Builder
setUnknownFields (com.google.protobuf.UnknownFieldSet unknownFields)
GPUOptions.Experimental.Builder
setUseUnifiedMemory (boolean value)
 If true, uses CUDA unified memory for memory allocations.
GPUOptions.Experimental.Builder
setVirtualDevices (int index, GPUOptions.Experimental.VirtualDevices.Builder builderForValue)
 The multi virtual device settings.
GPUOptions.Experimental.Builder
setVirtualDevices (int index, GPUOptions.Experimental.VirtualDevices value)
 The multi virtual device settings.

Inherited Methods

Public Methods

public GPUOptions.Experimental.Builder addAllVirtualDevices (Iterable<? extends GPUOptions.Experimental.VirtualDevices > values)

 The multi virtual device settings. If empty (not set), it will create
 single virtual device on each visible GPU, according to the settings
 in "visible_device_list" above. Otherwise, the number of elements in the
 list must be the same as the number of visible GPUs (after
 "visible_device_list" filtering if it is set), and the string represented
 device names (e.g. /device:GPU:<id>) will refer to the virtual
 devices and have the <id> field assigned sequentially starting from 0,
 according to the order they appear in this list and the "memory_limit"
 list inside each element. For example,
   visible_device_list = "1,0"
   virtual_devices { memory_limit: 1GB memory_limit: 2GB }
   virtual_devices {}
 will create three virtual devices as:
   /device:GPU:0 -> visible GPU 1 with 1GB memory
   /device:GPU:1 -> visible GPU 1 with 2GB memory
   /device:GPU:2 -> visible GPU 0 with all available memory
 NOTE:
 1. It's invalid to set both this and "per_process_gpu_memory_fraction"
    at the same time.
 2. Currently this setting is per-process, not per-session. Using
    different settings in different sessions within same process will
    result in undefined behavior.
 
repeated .tensorflow.GPUOptions.Experimental.VirtualDevices virtual_devices = 1;

public