TensorFlow is back at Google I/O on May 14! Register now

GPUOptions.Experimental.VirtualDevices

public static final class GPUOptions.Experimental.VirtualDevices

 Configuration for breaking down a visible GPU into multiple "virtual"
 devices.

Protobuf type tensorflow.GPUOptions.Experimental.VirtualDevices

Nested Classes

class

GPUOptions.Experimental.VirtualDevices.Builder

 Configuration for breaking down a visible GPU into multiple "virtual"
 devices.

Constants

int	MEMORY_LIMIT_MB_FIELD_NUMBER
int	PRIORITY_FIELD_NUMBER

Public Methods

boolean	equals(Object obj)
static GPUOptions.Experimental.VirtualDevices	getDefaultInstance()
GPUOptions.Experimental.VirtualDevices	getDefaultInstanceForType()
final static com.google.protobuf.Descriptors.Descriptor	getDescriptor()
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.
	getParserForType()
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.
int	getSerializedSize()
final com.google.protobuf.UnknownFieldSet	getUnknownFields()
int	hashCode()
final boolean	isInitialized()
static GPUOptions.Experimental.VirtualDevices.Builder	newBuilder()
static GPUOptions.Experimental.VirtualDevices.Builder	newBuilder(GPUOptions.Experimental.VirtualDevices prototype)
GPUOptions.Experimental.VirtualDevices.Builder	newBuilderForType()
static GPUOptions.Experimental.VirtualDevices	parseDelimitedFrom(InputStream input)
static GPUOptions.Experimental.VirtualDevices	parseDelimitedFrom(InputStream input, com.google.protobuf.ExtensionRegistryLite extensionRegistry)
static GPUOptions.Experimental.VirtualDevices	parseFrom(ByteBuffer data)
static GPUOptions.Experimental.VirtualDevices	parseFrom(com.google.protobuf.CodedInputStream input, com.google.protobuf.ExtensionRegistryLite extensionRegistry)
static GPUOptions.Experimental.VirtualDevices	parseFrom(ByteBuffer data, com.google.protobuf.ExtensionRegistryLite extensionRegistry)
static GPUOptions.Experimental.VirtualDevices	parseFrom(com.google.protobuf.CodedInputStream input)
static GPUOptions.Experimental.VirtualDevices	parseFrom(byte[] data, com.google.protobuf.ExtensionRegistryLite extensionRegistry)
static GPUOptions.Experimental.VirtualDevices	parseFrom(com.google.protobuf.ByteString data)
static GPUOptions.Experimental.VirtualDevices	parseFrom(InputStream input, com.google.protobuf.ExtensionRegistryLite extensionRegistry)
static GPUOptions.Experimental.VirtualDevices	parseFrom(com.google.protobuf.ByteString data, com.google.protobuf.ExtensionRegistryLite extensionRegistry)
static	parser()
GPUOptions.Experimental.VirtualDevices.Builder	toBuilder()
void	writeTo(com.google.protobuf.CodedOutputStream output)

Inherited Methods

From class java.lang.Object

boolean	equals(Object arg0)
final Class<?>	getClass()
int	hashCode()
final void	notify()
final void	notifyAll()
String	toString()
final void	wait(long arg0, int arg1)
final void	wait(long arg0)
final void	wait()

From interface org.tensorflow.proto.framework.GPUOptions.Experimental.VirtualDevicesOrBuilder

abstract float	getMemoryLimitMb(int index) Per "virtual" device memory limit, in MB.
abstract int	getMemoryLimitMbCount() Per "virtual" device memory limit, in MB.
abstract List<Float>	getMemoryLimitMbList() Per "virtual" device memory limit, in MB.
abstract int	getPriority(int index) Priority values to use with the virtual devices.
abstract int	getPriorityCount() Priority values to use with the virtual devices.
abstract List<Integer>	getPriorityList() Priority values to use with the virtual devices.

Constants

public static final int MEMORY_LIMIT_MB_FIELD_NUMBER

Constant Value: 1

public static final int PRIORITY_FIELD_NUMBER

Constant Value: 2

Public Methods

public boolean equals (Object obj)

public static GPUOptions.Experimental.VirtualDevices getDefaultInstance ()

public GPUOptions.Experimental.VirtualDevices getDefaultInstanceForType ()

public static final com.google.protobuf.Descriptors.Descriptor getDescriptor ()

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 getParserForType ()

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 int getSerializedSize ()

public final com.google.protobuf.UnknownFieldSet getUnknownFields ()

public int hashCode ()

public final boolean isInitialized ()

public static GPUOptions.Experimental.VirtualDevices.Builder newBuilder ()

public static GPUOptions.Experimental.VirtualDevices.Builder newBuilder (GPUOptions.Experimental.VirtualDevices prototype)

public GPUOptions.Experimental.VirtualDevices.Builder newBuilderForType ()

public static GPUOptions.Experimental.VirtualDevices parseDelimitedFrom (InputStream input)

Throws

IOException

public static GPUOptions.Experimental.VirtualDevices parseDelimitedFrom (InputStream input, com.google.protobuf.ExtensionRegistryLite extensionRegistry)

Throws

IOException

public static GPUOptions.Experimental.VirtualDevices parseFrom (ByteBuffer data)

Throws

InvalidProtocolBufferException

public static GPUOptions.Experimental.VirtualDevices parseFrom (com.google.protobuf.CodedInputStream input, com.google.protobuf.ExtensionRegistryLite extensionRegistry)

Throws

IOException

public static GPUOptions.Experimental.VirtualDevices parseFrom (ByteBuffer data, com.google.protobuf.ExtensionRegistryLite extensionRegistry)

Throws

InvalidProtocolBufferException

public static GPUOptions.Experimental.VirtualDevices parseFrom (com.google.protobuf.CodedInputStream input)

Throws

IOException

public static GPUOptions.Experimental.VirtualDevices parseFrom (byte[] data, com.google.protobuf.ExtensionRegistryLite extensionRegistry)

Throws

InvalidProtocolBufferException

public static GPUOptions.Experimental.VirtualDevices parseFrom (com.google.protobuf.ByteString data)

Throws

InvalidProtocolBufferException

public static GPUOptions.Experimental.VirtualDevices parseFrom (InputStream input, com.google.protobuf.ExtensionRegistryLite extensionRegistry)

Throws

IOException

public static GPUOptions.Experimental.VirtualDevices parseFrom (com.google.protobuf.ByteString data, com.google.protobuf.ExtensionRegistryLite extensionRegistry)

Throws

InvalidProtocolBufferException

public static parser ()

public GPUOptions.Experimental.VirtualDevices.Builder toBuilder ()

public void writeTo (com.google.protobuf.CodedOutputStream output)

Throws

IOException