public static final class
GPUOptions.Experimental.VirtualDevices.Builder
Configuration for breaking down a visible GPU into multiple "virtual" devices.
tensorflow.GPUOptions.Experimental.VirtualDevices
Public Methods
GPUOptions.Experimental.VirtualDevices.Builder |
addAllMemoryLimitMb(Iterable<? extends Float> values)
Per "virtual" device memory limit, in MB. |
GPUOptions.Experimental.VirtualDevices.Builder |
addAllPriority(Iterable<? extends Integer> values)
Priority values to use with the virtual devices. |
GPUOptions.Experimental.VirtualDevices.Builder |
addMemoryLimitMb(float value)
Per "virtual" device memory limit, in MB. |
GPUOptions.Experimental.VirtualDevices.Builder |
addPriority(int value)
Priority values to use with the virtual devices. |
GPUOptions.Experimental.VirtualDevices.Builder |
addRepeatedField(com.google.protobuf.Descriptors.FieldDescriptor field, Object value)
|
GPUOptions.Experimental.VirtualDevices |
build()
|
GPUOptions.Experimental.VirtualDevices | |
GPUOptions.Experimental.VirtualDevices.Builder |
clear()
|
GPUOptions.Experimental.VirtualDevices.Builder |
clearField(com.google.protobuf.Descriptors.FieldDescriptor field)
|
GPUOptions.Experimental.VirtualDevices.Builder |
clearMemoryLimitMb()
Per "virtual" device memory limit, in MB. |
GPUOptions.Experimental.VirtualDevices.Builder |
clearOneof(com.google.protobuf.Descriptors.OneofDescriptor oneof)
|
GPUOptions.Experimental.VirtualDevices.Builder |
clearPriority()
Priority values to use with the virtual devices. |
GPUOptions.Experimental.VirtualDevices.Builder |
clone()
|
GPUOptions.Experimental.VirtualDevices | |
final static com.google.protobuf.Descriptors.Descriptor | |
com.google.protobuf.Descriptors.Descriptor | |
float |
getMemoryLimitMb(int index)
Per "virtual" device memory limit, in MB. |
int |
getMemoryLimitMbCount()
Per "virtual" device memory limit, in MB. |
List<Float> |
getMemoryLimitMbList()
Per "virtual" device memory limit, in MB. |
int |
getPriority(int index)
Priority values to use with the virtual devices. |
int |
getPriorityCount()
Priority values to use with the virtual devices. |
List<Integer> |
getPriorityList()
Priority values to use with the virtual devices. |
final boolean | |
GPUOptions.Experimental.VirtualDevices.Builder |
mergeFrom(com.google.protobuf.CodedInputStream input, com.google.protobuf.ExtensionRegistryLite extensionRegistry)
|
GPUOptions.Experimental.VirtualDevices.Builder |
mergeFrom(com.google.protobuf.Message other)
|
final GPUOptions.Experimental.VirtualDevices.Builder |
mergeUnknownFields(com.google.protobuf.UnknownFieldSet unknownFields)
|
GPUOptions.Experimental.VirtualDevices.Builder |
setField(com.google.protobuf.Descriptors.FieldDescriptor field, Object value)
|
GPUOptions.Experimental.VirtualDevices.Builder |
setMemoryLimitMb(int index, float value)
Per "virtual" device memory limit, in MB. |
GPUOptions.Experimental.VirtualDevices.Builder |
setPriority(int index, int value)
Priority values to use with the virtual devices. |
GPUOptions.Experimental.VirtualDevices.Builder |
setRepeatedField(com.google.protobuf.Descriptors.FieldDescriptor field, int index, Object value)
|
final GPUOptions.Experimental.VirtualDevices.Builder |
setUnknownFields(com.google.protobuf.UnknownFieldSet unknownFields)
|
Inherited Methods
Public Methods
public GPUOptions.Experimental.VirtualDevices.Builder addAllMemoryLimitMb (Iterable<? extends Float> values)
Per "virtual" device memory limit, in MB. The number of elements in the list is the number of virtual devices to create on the corresponding visible GPU (see "virtual_devices" below). If empty, it will create single virtual device taking all available memory from the device. For the concept of "visible" and "virtual" GPU, see the comments for "visible_device_list" above for more information.
repeated float memory_limit_mb = 1;
public GPUOptions.Experimental.VirtualDevices.Builder addAllPriority (Iterable<? extends Integer> values)
Priority values to use with the virtual devices. Use the cuda function cudaDeviceGetStreamPriorityRange to query for valid range of values for priority. On a P4000 GPU with cuda 10.1, the priority range reported was 0 for least priority and -1 for greatest priority. If this field is not specified, then the virtual devices will be created with the default. If this field has values set, then the size of this must match with the above memory_limit_mb.
repeated int32 priority = 2;
public GPUOptions.Experimental.VirtualDevices.Builder addMemoryLimitMb (float value)
Per "virtual" device memory limit, in MB. The number of elements in the list is the number of virtual devices to create on the corresponding visible GPU (see "virtual_devices" below). If empty, it will create single virtual device taking all available memory from the device. For the concept of "visible" and "virtual" GPU, see the comments for "visible_device_list" above for more information.
repeated float memory_limit_mb = 1;
public GPUOptions.Experimental.VirtualDevices.Builder addPriority (int value)
Priority values to use with the virtual devices. Use the cuda function cudaDeviceGetStreamPriorityRange to query for valid range of values for priority. On a P4000 GPU with cuda 10.1, the priority range reported was 0 for least priority and -1 for greatest priority. If this field is not specified, then the virtual devices will be created with the default. If this field has values set, then the size of this must match with the above memory_limit_mb.
repeated int32 priority = 2;
public GPUOptions.Experimental.VirtualDevices.Builder addRepeatedField (com.google.protobuf.Descriptors.FieldDescriptor field, Object value)
public GPUOptions.Experimental.VirtualDevices.Builder clearField (com.google.protobuf.Descriptors.FieldDescriptor field)
public GPUOptions.Experimental.VirtualDevices.Builder clearMemoryLimitMb ()
Per "virtual" device memory limit, in MB. The number of elements in the list is the number of virtual devices to create on the corresponding visible GPU (see "virtual_devices" below). If empty, it will create single virtual device taking all available memory from the device. For the concept of "visible" and "virtual" GPU, see the comments for "visible_device_list" above for more information.
repeated float memory_limit_mb = 1;
public GPUOptions.Experimental.VirtualDevices.Builder clearOneof (com.google.protobuf.Descriptors.OneofDescriptor oneof)
public GPUOptions.Experimental.VirtualDevices.Builder clearPriority ()
Priority values to use with the virtual devices. Use the cuda function cudaDeviceGetStreamPriorityRange to query for valid range of values for priority. On a P4000 GPU with cuda 10.1, the priority range reported was 0 for least priority and -1 for greatest priority. If this field is not specified, then the virtual devices will be created with the default. If this field has values set, then the size of this must match with the above memory_limit_mb.
repeated int32 priority = 2;
public static final com.google.protobuf.Descriptors.Descriptor getDescriptor ()
public com.google.protobuf.Descriptors.Descriptor getDescriptorForType ()
public float getMemoryLimitMb (int index)
Per "virtual" device memory limit, in MB. The number of elements in the list is the number of virtual devices to create on the corresponding visible GPU (see "virtual_devices" below). If empty, it will create single virtual device taking all available memory from the device. For the concept of "visible" and "virtual" GPU, see the comments for "visible_device_list" above for more information.
repeated float memory_limit_mb = 1;
public int getMemoryLimitMbCount ()
Per "virtual" device memory limit, in MB. The number of elements in the list is the number of virtual devices to create on the corresponding visible GPU (see "virtual_devices" below). If empty, it will create single virtual device taking all available memory from the device. For the concept of "visible" and "virtual" GPU, see the comments for "visible_device_list" above for more information.
repeated float memory_limit_mb = 1;
public List<Float> getMemoryLimitMbList ()
Per "virtual" device memory limit, in MB. The number of elements in the list is the number of virtual devices to create on the corresponding visible GPU (see "virtual_devices" below). If empty, it will create single virtual device taking all available memory from the device. For the concept of "visible" and "virtual" GPU, see the comments for "visible_device_list" above for more information.
repeated float memory_limit_mb = 1;
public int getPriority (int index)
Priority values to use with the virtual devices. Use the cuda function cudaDeviceGetStreamPriorityRange to query for valid range of values for priority. On a P4000 GPU with cuda 10.1, the priority range reported was 0 for least priority and -1 for greatest priority. If this field is not specified, then the virtual devices will be created with the default. If this field has values set, then the size of this must match with the above memory_limit_mb.
repeated int32 priority = 2;
public int getPriorityCount ()
Priority values to use with the virtual devices. Use the cuda function cudaDeviceGetStreamPriorityRange to query for valid range of values for priority. On a P4000 GPU with cuda 10.1, the priority range reported was 0 for least priority and -1 for greatest priority. If this field is not specified, then the virtual devices will be created with the default. If this field has values set, then the size of this must match with the above memory_limit_mb.
repeated int32 priority = 2;
public List<Integer> getPriorityList ()
Priority values to use with the virtual devices. Use the cuda function cudaDeviceGetStreamPriorityRange to query for valid range of values for priority. On a P4000 GPU with cuda 10.1, the priority range reported was 0 for least priority and -1 for greatest priority. If this field is not specified, then the virtual devices will be created with the default. If this field has values set, then the size of this must match with the above memory_limit_mb.
repeated int32 priority = 2;
public final boolean isInitialized ()
public GPUOptions.Experimental.VirtualDevices.Builder mergeFrom (com.google.protobuf.CodedInputStream input, com.google.protobuf.ExtensionRegistryLite extensionRegistry)
Throws
IOException |
---|
public final GPUOptions.Experimental.VirtualDevices.Builder mergeUnknownFields (com.google.protobuf.UnknownFieldSet unknownFields)
public GPUOptions.Experimental.VirtualDevices.Builder setField (com.google.protobuf.Descriptors.FieldDescriptor field, Object value)
public GPUOptions.Experimental.VirtualDevices.Builder setMemoryLimitMb (int index, float value)
Per "virtual" device memory limit, in MB. The number of elements in the list is the number of virtual devices to create on the corresponding visible GPU (see "virtual_devices" below). If empty, it will create single virtual device taking all available memory from the device. For the concept of "visible" and "virtual" GPU, see the comments for "visible_device_list" above for more information.
repeated float memory_limit_mb = 1;
public GPUOptions.Experimental.VirtualDevices.Builder setPriority (int index, int value)
Priority values to use with the virtual devices. Use the cuda function cudaDeviceGetStreamPriorityRange to query for valid range of values for priority. On a P4000 GPU with cuda 10.1, the priority range reported was 0 for least priority and -1 for greatest priority. If this field is not specified, then the virtual devices will be created with the default. If this field has values set, then the size of this must match with the above memory_limit_mb.
repeated int32 priority = 2;