Package org.tensorflow.framework
Class GPUOptions.Experimental.Builder
java.lang.Object
com.google.protobuf.AbstractMessageLite.Builder
com.google.protobuf.AbstractMessage.Builder<BuilderType>
com.google.protobuf.GeneratedMessageV3.Builder<GPUOptions.Experimental.Builder>
org.tensorflow.framework.GPUOptions.Experimental.Builder
- All Implemented Interfaces:
com.google.protobuf.Message.Builder,com.google.protobuf.MessageLite.Builder,com.google.protobuf.MessageLiteOrBuilder,com.google.protobuf.MessageOrBuilder,java.lang.Cloneable,GPUOptions.ExperimentalOrBuilder
- Enclosing class:
- GPUOptions.Experimental
public static final class GPUOptions.Experimental.Builder extends com.google.protobuf.GeneratedMessageV3.Builder<GPUOptions.Experimental.Builder> implements GPUOptions.ExperimentalOrBuilder
Protobuf type
tensorflow.GPUOptions.Experimental-
Method Summary
Modifier and Type Method Description GPUOptions.Experimental.BuilderaddAllVirtualDevices(java.lang.Iterable<? extends GPUOptions.Experimental.VirtualDevices> values)The multi virtual device settings.GPUOptions.Experimental.BuilderaddRepeatedField(com.google.protobuf.Descriptors.FieldDescriptor field, java.lang.Object value)GPUOptions.Experimental.BuilderaddVirtualDevices(int index, GPUOptions.Experimental.VirtualDevices value)The multi virtual device settings.GPUOptions.Experimental.BuilderaddVirtualDevices(int index, GPUOptions.Experimental.VirtualDevices.Builder builderForValue)The multi virtual device settings.GPUOptions.Experimental.BuilderaddVirtualDevices(GPUOptions.Experimental.VirtualDevices value)The multi virtual device settings.GPUOptions.Experimental.BuilderaddVirtualDevices(GPUOptions.Experimental.VirtualDevices.Builder builderForValue)The multi virtual device settings.GPUOptions.Experimental.VirtualDevices.BuilderaddVirtualDevicesBuilder()The multi virtual device settings.GPUOptions.Experimental.VirtualDevices.BuilderaddVirtualDevicesBuilder(int index)The multi virtual device settings.GPUOptions.Experimentalbuild()GPUOptions.ExperimentalbuildPartial()GPUOptions.Experimental.Builderclear()GPUOptions.Experimental.BuilderclearCollectiveRingOrder()If non-empty, defines a good GPU ring order on a single worker based on device interconnect.GPUOptions.Experimental.BuilderclearField(com.google.protobuf.Descriptors.FieldDescriptor field)GPUOptions.Experimental.BuilderclearKernelTrackerMaxBytes()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.BuilderclearKernelTrackerMaxInterval()Parameters for GPUKernelTracker.GPUOptions.Experimental.BuilderclearKernelTrackerMaxPending()If kernel_tracker_max_pending > 0 then no more than this many tracking events can be outstanding at a time.GPUOptions.Experimental.BuilderclearNumDevToDevCopyStreams()If > 1, the number of device-to-device copy streams to create for each GPUDevice.GPUOptions.Experimental.BuilderclearOneof(com.google.protobuf.Descriptors.OneofDescriptor oneof)GPUOptions.Experimental.BuilderclearTimestampedAllocator()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.BuilderclearUseUnifiedMemory()If true, uses CUDA unified memory for memory allocations.GPUOptions.Experimental.BuilderclearVirtualDevices()The multi virtual device settings.GPUOptions.Experimental.Builderclone()java.lang.StringgetCollectiveRingOrder()If non-empty, defines a good GPU ring order on a single worker based on device interconnect.com.google.protobuf.ByteStringgetCollectiveRingOrderBytes()If non-empty, defines a good GPU ring order on a single worker based on device interconnect.GPUOptions.ExperimentalgetDefaultInstanceForType()static com.google.protobuf.Descriptors.DescriptorgetDescriptor()com.google.protobuf.Descriptors.DescriptorgetDescriptorForType()intgetKernelTrackerMaxBytes()If kernel_tracker_max_bytes = n > 0, then a tracking event is inserted after every series of kernels allocating a sum of memory >= n.intgetKernelTrackerMaxInterval()Parameters for GPUKernelTracker.intgetKernelTrackerMaxPending()If kernel_tracker_max_pending > 0 then no more than this many tracking events can be outstanding at a time.intgetNumDevToDevCopyStreams()If > 1, the number of device-to-device copy streams to create for each GPUDevice.booleangetTimestampedAllocator()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.booleangetUseUnifiedMemory()If true, uses CUDA unified memory for memory allocations.GPUOptions.Experimental.VirtualDevicesgetVirtualDevices(int index)The multi virtual device settings.GPUOptions.Experimental.VirtualDevices.BuildergetVirtualDevicesBuilder(int index)The multi virtual device settings.java.util.List<GPUOptions.Experimental.VirtualDevices.Builder>getVirtualDevicesBuilderList()The multi virtual device settings.intgetVirtualDevicesCount()The multi virtual device settings.java.util.List<GPUOptions.Experimental.VirtualDevices>getVirtualDevicesList()The multi virtual device settings.GPUOptions.Experimental.VirtualDevicesOrBuildergetVirtualDevicesOrBuilder(int index)The multi virtual device settings.java.util.List<? extends GPUOptions.Experimental.VirtualDevicesOrBuilder>getVirtualDevicesOrBuilderList()The multi virtual device settings.protected com.google.protobuf.GeneratedMessageV3.FieldAccessorTableinternalGetFieldAccessorTable()booleanisInitialized()GPUOptions.Experimental.BuildermergeFrom(com.google.protobuf.CodedInputStream input, com.google.protobuf.ExtensionRegistryLite extensionRegistry)GPUOptions.Experimental.BuildermergeFrom(com.google.protobuf.Message other)GPUOptions.Experimental.BuildermergeFrom(GPUOptions.Experimental other)GPUOptions.Experimental.BuildermergeUnknownFields(com.google.protobuf.UnknownFieldSet unknownFields)GPUOptions.Experimental.BuilderremoveVirtualDevices(int index)The multi virtual device settings.GPUOptions.Experimental.BuildersetCollectiveRingOrder(java.lang.String value)If non-empty, defines a good GPU ring order on a single worker based on device interconnect.GPUOptions.Experimental.BuildersetCollectiveRingOrderBytes(com.google.protobuf.ByteString value)If non-empty, defines a good GPU ring order on a single worker based on device interconnect.GPUOptions.Experimental.BuildersetField(com.google.protobuf.Descriptors.FieldDescriptor field, java.lang.Object value)GPUOptions.Experimental.BuildersetKernelTrackerMaxBytes(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.BuildersetKernelTrackerMaxInterval(int value)Parameters for GPUKernelTracker.GPUOptions.Experimental.BuildersetKernelTrackerMaxPending(int value)If kernel_tracker_max_pending > 0 then no more than this many tracking events can be outstanding at a time.GPUOptions.Experimental.BuildersetNumDevToDevCopyStreams(int value)If > 1, the number of device-to-device copy streams to create for each GPUDevice.GPUOptions.Experimental.BuildersetRepeatedField(com.google.protobuf.Descriptors.FieldDescriptor field, int index, java.lang.Object value)GPUOptions.Experimental.BuildersetTimestampedAllocator(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.GPUOptions.Experimental.BuildersetUnknownFields(com.google.protobuf.UnknownFieldSet unknownFields)GPUOptions.Experimental.BuildersetUseUnifiedMemory(boolean value)If true, uses CUDA unified memory for memory allocations.GPUOptions.Experimental.BuildersetVirtualDevices(int index, GPUOptions.Experimental.VirtualDevices value)The multi virtual device settings.GPUOptions.Experimental.BuildersetVirtualDevices(int index, GPUOptions.Experimental.VirtualDevices.Builder builderForValue)The multi virtual device settings.Methods inherited from class com.google.protobuf.GeneratedMessageV3.Builder
getAllFields, getField, getFieldBuilder, getOneofFieldDescriptor, getParentForChildren, getRepeatedField, getRepeatedFieldBuilder, getRepeatedFieldCount, getUnknownFields, hasField, hasOneof, internalGetMapField, internalGetMutableMapField, isClean, markClean, newBuilderForField, onBuilt, onChanged, setUnknownFieldsProto3Methods inherited from class com.google.protobuf.AbstractMessage.Builder
findInitializationErrors, getInitializationErrorString, internalMergeFrom, mergeDelimitedFrom, mergeDelimitedFrom, mergeFrom, mergeFrom, mergeFrom, mergeFrom, mergeFrom, mergeFrom, mergeFrom, mergeFrom, mergeFrom, newUninitializedMessageException, toStringMethods inherited from class com.google.protobuf.AbstractMessageLite.Builder
addAll, addAll, mergeFrom, newUninitializedMessageException
-
Method Details
-
getDescriptor
public static final com.google.protobuf.Descriptors.Descriptor getDescriptor() -
internalGetFieldAccessorTable
protected com.google.protobuf.GeneratedMessageV3.FieldAccessorTable internalGetFieldAccessorTable()- Specified by:
internalGetFieldAccessorTablein classcom.google.protobuf.GeneratedMessageV3.Builder<GPUOptions.Experimental.Builder>
-
clear
- Specified by:
clearin interfacecom.google.protobuf.Message.Builder- Specified by:
clearin interfacecom.google.protobuf.MessageLite.Builder- Overrides:
clearin classcom.google.protobuf.GeneratedMessageV3.Builder<GPUOptions.Experimental.Builder>
-
getDescriptorForType
public com.google.protobuf.Descriptors.Descriptor getDescriptorForType()- Specified by:
getDescriptorForTypein interfacecom.google.protobuf.Message.Builder- Specified by:
getDescriptorForTypein interfacecom.google.protobuf.MessageOrBuilder- Overrides:
getDescriptorForTypein classcom.google.protobuf.GeneratedMessageV3.Builder<GPUOptions.Experimental.Builder>
-
getDefaultInstanceForType
- Specified by:
getDefaultInstanceForTypein interfacecom.google.protobuf.MessageLiteOrBuilder- Specified by:
getDefaultInstanceForTypein interfacecom.google.protobuf.MessageOrBuilder
-
build
- Specified by:
buildin interfacecom.google.protobuf.Message.Builder- Specified by:
buildin interfacecom.google.protobuf.MessageLite.Builder
-
buildPartial
- Specified by:
buildPartialin interfacecom.google.protobuf.Message.Builder- Specified by:
buildPartialin interfacecom.google.protobuf.MessageLite.Builder
-
clone
- Specified by:
clonein interfacecom.google.protobuf.Message.Builder- Specified by:
clonein interfacecom.google.protobuf.MessageLite.Builder- Overrides:
clonein classcom.google.protobuf.GeneratedMessageV3.Builder<GPUOptions.Experimental.Builder>
-
setField
public GPUOptions.Experimental.Builder setField(com.google.protobuf.Descriptors.FieldDescriptor field, java.lang.Object value)- Specified by:
setFieldin interfacecom.google.protobuf.Message.Builder- Overrides:
setFieldin classcom.google.protobuf.GeneratedMessageV3.Builder<GPUOptions.Experimental.Builder>
-
clearField
public GPUOptions.Experimental.Builder clearField(com.google.protobuf.Descriptors.FieldDescriptor field)- Specified by:
clearFieldin interfacecom.google.protobuf.Message.Builder- Overrides:
clearFieldin classcom.google.protobuf.GeneratedMessageV3.Builder<GPUOptions.Experimental.Builder>
-
clearOneof
public GPUOptions.Experimental.Builder clearOneof(com.google.protobuf.Descriptors.OneofDescriptor oneof)- Specified by:
clearOneofin interfacecom.google.protobuf.Message.Builder- Overrides:
clearOneofin classcom.google.protobuf.GeneratedMessageV3.Builder<GPUOptions.Experimental.Builder>
-
setRepeatedField
public GPUOptions.Experimental.Builder setRepeatedField(com.google.protobuf.Descriptors.FieldDescriptor field, int index, java.lang.Object value)- Specified by:
setRepeatedFieldin interfacecom.google.protobuf.Message.Builder- Overrides:
setRepeatedFieldin classcom.google.protobuf.GeneratedMessageV3.Builder<GPUOptions.Experimental.Builder>
-
addRepeatedField
public GPUOptions.Experimental.Builder addRepeatedField(com.google.protobuf.Descriptors.FieldDescriptor field, java.lang.Object value)- Specified by:
addRepeatedFieldin interfacecom.google.protobuf.Message.Builder- Overrides:
addRepeatedFieldin classcom.google.protobuf.GeneratedMessageV3.Builder<GPUOptions.Experimental.Builder>
-
mergeFrom
- Specified by:
mergeFromin interfacecom.google.protobuf.Message.Builder- Overrides:
mergeFromin classcom.google.protobuf.AbstractMessage.Builder<GPUOptions.Experimental.Builder>
-
mergeFrom
-
isInitialized
public final boolean isInitialized()- Specified by:
isInitializedin interfacecom.google.protobuf.MessageLiteOrBuilder- Overrides:
isInitializedin classcom.google.protobuf.GeneratedMessageV3.Builder<GPUOptions.Experimental.Builder>
-
mergeFrom
public GPUOptions.Experimental.Builder mergeFrom(com.google.protobuf.CodedInputStream input, com.google.protobuf.ExtensionRegistryLite extensionRegistry) throws java.io.IOException- Specified by:
mergeFromin interfacecom.google.protobuf.Message.Builder- Specified by:
mergeFromin interfacecom.google.protobuf.MessageLite.Builder- Overrides:
mergeFromin classcom.google.protobuf.AbstractMessage.Builder<GPUOptions.Experimental.Builder>- Throws:
java.io.IOException
-
getVirtualDevicesList
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;- Specified by:
getVirtualDevicesListin interfaceGPUOptions.ExperimentalOrBuilder
-
getVirtualDevicesCount
public int getVirtualDevicesCount()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;- Specified by:
getVirtualDevicesCountin interfaceGPUOptions.ExperimentalOrBuilder
-
getVirtualDevices
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;- Specified by:
getVirtualDevicesin interfaceGPUOptions.ExperimentalOrBuilder
-
setVirtualDevices
public GPUOptions.Experimental.Builder setVirtualDevices(int index, GPUOptions.Experimental.VirtualDevices value)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; -
setVirtualDevices
public GPUOptions.Experimental.Builder setVirtualDevices(int index, GPUOptions.Experimental.VirtualDevices.Builder builderForValue)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; -
addVirtualDevices
public GPUOptions.Experimental.Builder addVirtualDevices(GPUOptions.Experimental.VirtualDevices value)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; -
addVirtualDevices
public GPUOptions.Experimental.Builder addVirtualDevices(int index, GPUOptions.Experimental.VirtualDevices value)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; -
addVirtualDevices
public GPUOptions.Experimental.Builder addVirtualDevices(GPUOptions.Experimental.VirtualDevices.Builder builderForValue)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; -
addVirtualDevices
public GPUOptions.Experimental.Builder addVirtualDevices(int index, GPUOptions.Experimental.VirtualDevices.Builder builderForValue)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; -
addAllVirtualDevices
public GPUOptions.Experimental.Builder addAllVirtualDevices(java.lang.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; -
clearVirtualDevices
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; -
removeVirtualDevices
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; -
getVirtualDevicesBuilder
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; -
getVirtualDevicesOrBuilder
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;- Specified by:
getVirtualDevicesOrBuilderin interfaceGPUOptions.ExperimentalOrBuilder
-
getVirtualDevicesOrBuilderList
public java.util.List<? extends GPUOptions.Experimental.VirtualDevicesOrBuilder> getVirtualDevicesOrBuilderList()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;- Specified by:
getVirtualDevicesOrBuilderListin interfaceGPUOptions.ExperimentalOrBuilder
-
addVirtualDevicesBuilder
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; -
addVirtualDevicesBuilder
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; -
getVirtualDevicesBuilderList
public java.util.List<GPUOptions.Experimental.VirtualDevices.Builder> getVirtualDevicesBuilderList()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; -
getUseUnifiedMemory
public boolean getUseUnifiedMemory()If true, uses CUDA unified memory for memory allocations. If per_process_gpu_memory_fraction option is greater than 1.0, then unified memory is used regardless of the value for this field. See comments for per_process_gpu_memory_fraction field for more details and requirements of the unified memory. This option is useful to oversubscribe memory if multiple processes are sharing a single GPU while individually using less than 1.0 per process memory fraction.
bool use_unified_memory = 2;- Specified by:
getUseUnifiedMemoryin interfaceGPUOptions.ExperimentalOrBuilder- Returns:
- The useUnifiedMemory.
-
setUseUnifiedMemory
If true, uses CUDA unified memory for memory allocations. If per_process_gpu_memory_fraction option is greater than 1.0, then unified memory is used regardless of the value for this field. See comments for per_process_gpu_memory_fraction field for more details and requirements of the unified memory. This option is useful to oversubscribe memory if multiple processes are sharing a single GPU while individually using less than 1.0 per process memory fraction.
bool use_unified_memory = 2;- Parameters:
value- The useUnifiedMemory to set.- Returns:
- This builder for chaining.
-
clearUseUnifiedMemory
If true, uses CUDA unified memory for memory allocations. If per_process_gpu_memory_fraction option is greater than 1.0, then unified memory is used regardless of the value for this field. See comments for per_process_gpu_memory_fraction field for more details and requirements of the unified memory. This option is useful to oversubscribe memory if multiple processes are sharing a single GPU while individually using less than 1.0 per process memory fraction.
bool use_unified_memory = 2;- Returns:
- This builder for chaining.
-
getNumDevToDevCopyStreams
public int getNumDevToDevCopyStreams()If > 1, the number of device-to-device copy streams to create for each GPUDevice. Default value is 0, which is automatically converted to 1.
int32 num_dev_to_dev_copy_streams = 3;- Specified by:
getNumDevToDevCopyStreamsin interfaceGPUOptions.ExperimentalOrBuilder- Returns:
- The numDevToDevCopyStreams.
-
setNumDevToDevCopyStreams
If > 1, the number of device-to-device copy streams to create for each GPUDevice. Default value is 0, which is automatically converted to 1.
int32 num_dev_to_dev_copy_streams = 3;- Parameters:
value- The numDevToDevCopyStreams to set.- Returns:
- This builder for chaining.
-
clearNumDevToDevCopyStreams
If > 1, the number of device-to-device copy streams to create for each GPUDevice. Default value is 0, which is automatically converted to 1.
int32 num_dev_to_dev_copy_streams = 3;- Returns:
- This builder for chaining.
-
getCollectiveRingOrder
public java.lang.String getCollectiveRingOrder()If non-empty, defines a good GPU ring order on a single worker based on device interconnect. This assumes that all workers have the same GPU topology. Specify as a comma-separated string, e.g. "3,2,1,0,7,6,5,4". This ring order is used by the RingReducer implementation of CollectiveReduce, and serves as an override to automatic ring order generation in OrderTaskDeviceMap() during CollectiveParam resolution.
string collective_ring_order = 4;- Specified by:
getCollectiveRingOrderin interfaceGPUOptions.ExperimentalOrBuilder- Returns:
- The collectiveRingOrder.
-
getCollectiveRingOrderBytes
public com.google.protobuf.ByteString getCollectiveRingOrderBytes()If non-empty, defines a good GPU ring order on a single worker based on device interconnect. This assumes that all workers have the same GPU topology. Specify as a comma-separated string, e.g. "3,2,1,0,7,6,5,4". This ring order is used by the RingReducer implementation of CollectiveReduce, and serves as an override to automatic ring order generation in OrderTaskDeviceMap() during CollectiveParam resolution.
string collective_ring_order = 4;- Specified by:
getCollectiveRingOrderBytesin interfaceGPUOptions.ExperimentalOrBuilder- Returns:
- The bytes for collectiveRingOrder.
-
setCollectiveRingOrder
If non-empty, defines a good GPU ring order on a single worker based on device interconnect. This assumes that all workers have the same GPU topology. Specify as a comma-separated string, e.g. "3,2,1,0,7,6,5,4". This ring order is used by the RingReducer implementation of CollectiveReduce, and serves as an override to automatic ring order generation in OrderTaskDeviceMap() during CollectiveParam resolution.
string collective_ring_order = 4;- Parameters:
value- The collectiveRingOrder to set.- Returns:
- This builder for chaining.
-
clearCollectiveRingOrder
If non-empty, defines a good GPU ring order on a single worker based on device interconnect. This assumes that all workers have the same GPU topology. Specify as a comma-separated string, e.g. "3,2,1,0,7,6,5,4". This ring order is used by the RingReducer implementation of CollectiveReduce, and serves as an override to automatic ring order generation in OrderTaskDeviceMap() during CollectiveParam resolution.
string collective_ring_order = 4;- Returns:
- This builder for chaining.
-
setCollectiveRingOrderBytes
public 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. This assumes that all workers have the same GPU topology. Specify as a comma-separated string, e.g. "3,2,1,0,7,6,5,4". This ring order is used by the RingReducer implementation of CollectiveReduce, and serves as an override to automatic ring order generation in OrderTaskDeviceMap() during CollectiveParam resolution.
string collective_ring_order = 4;- Parameters:
value- The bytes for collectiveRingOrder to set.- Returns:
- This builder for chaining.
-
getTimestampedAllocator
public 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.
bool timestamped_allocator = 5;- Specified by:
getTimestampedAllocatorin interfaceGPUOptions.ExperimentalOrBuilder- Returns:
- The timestampedAllocator.
-
setTimestampedAllocator
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.
bool timestamped_allocator = 5;- Parameters:
value- The timestampedAllocator to set.- Returns:
- This builder for chaining.
-
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.
bool timestamped_allocator = 5;- Returns:
- This builder for chaining.
-
getKernelTrackerMaxInterval
public int getKernelTrackerMaxInterval()Parameters for GPUKernelTracker. By default no kernel tracking is done. Note that timestamped_allocator is only effective if some tracking is specified. If kernel_tracker_max_interval = n > 0, then a tracking event is inserted after every n kernels without an event.
int32 kernel_tracker_max_interval = 7;- Specified by:
getKernelTrackerMaxIntervalin interfaceGPUOptions.ExperimentalOrBuilder- Returns:
- The kernelTrackerMaxInterval.
-
setKernelTrackerMaxInterval
Parameters for GPUKernelTracker. By default no kernel tracking is done. Note that timestamped_allocator is only effective if some tracking is specified. If kernel_tracker_max_interval = n > 0, then a tracking event is inserted after every n kernels without an event.
int32 kernel_tracker_max_interval = 7;- Parameters:
value- The kernelTrackerMaxInterval to set.- Returns:
- This builder for chaining.
-
clearKernelTrackerMaxInterval
Parameters for GPUKernelTracker. By default no kernel tracking is done. Note that timestamped_allocator is only effective if some tracking is specified. If kernel_tracker_max_interval = n > 0, then a tracking event is inserted after every n kernels without an event.
int32 kernel_tracker_max_interval = 7;- Returns:
- This builder for chaining.
-
getKernelTrackerMaxBytes
public 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. If one kernel allocates b * n bytes, then one event will be inserted after it, but it will count as b against the pending limit.
int32 kernel_tracker_max_bytes = 8;- Specified by:
getKernelTrackerMaxBytesin interfaceGPUOptions.ExperimentalOrBuilder- Returns:
- The kernelTrackerMaxBytes.
-
setKernelTrackerMaxBytes
If kernel_tracker_max_bytes = n > 0, then a tracking event is inserted after every series of kernels allocating a sum of memory >= n. If one kernel allocates b * n bytes, then one event will be inserted after it, but it will count as b against the pending limit.
int32 kernel_tracker_max_bytes = 8;- Parameters:
value- The kernelTrackerMaxBytes to set.- Returns:
- This builder for chaining.
-
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. If one kernel allocates b * n bytes, then one event will be inserted after it, but it will count as b against the pending limit.
int32 kernel_tracker_max_bytes = 8;- Returns:
- This builder for chaining.
-
getKernelTrackerMaxPending
public int getKernelTrackerMaxPending()If kernel_tracker_max_pending > 0 then no more than this many tracking events can be outstanding at a time. An attempt to launch an additional kernel will stall until an event completes.
int32 kernel_tracker_max_pending = 9;- Specified by:
getKernelTrackerMaxPendingin interfaceGPUOptions.ExperimentalOrBuilder- Returns:
- The kernelTrackerMaxPending.
-
setKernelTrackerMaxPending
If kernel_tracker_max_pending > 0 then no more than this many tracking events can be outstanding at a time. An attempt to launch an additional kernel will stall until an event completes.
int32 kernel_tracker_max_pending = 9;- Parameters:
value- The kernelTrackerMaxPending to set.- Returns:
- This builder for chaining.
-
clearKernelTrackerMaxPending
If kernel_tracker_max_pending > 0 then no more than this many tracking events can be outstanding at a time. An attempt to launch an additional kernel will stall until an event completes.
int32 kernel_tracker_max_pending = 9;- Returns:
- This builder for chaining.
-
setUnknownFields
public final GPUOptions.Experimental.Builder setUnknownFields(com.google.protobuf.UnknownFieldSet unknownFields)- Specified by:
setUnknownFieldsin interfacecom.google.protobuf.Message.Builder- Overrides:
setUnknownFieldsin classcom.google.protobuf.GeneratedMessageV3.Builder<GPUOptions.Experimental.Builder>
-
mergeUnknownFields
public final GPUOptions.Experimental.Builder mergeUnknownFields(com.google.protobuf.UnknownFieldSet unknownFields)- Specified by:
mergeUnknownFieldsin interfacecom.google.protobuf.Message.Builder- Overrides:
mergeUnknownFieldsin classcom.google.protobuf.GeneratedMessageV3.Builder<GPUOptions.Experimental.Builder>
-