Package org.tensorflow.framework
Interface GPUOptions.Experimental.VirtualDevicesOrBuilder
- All Superinterfaces:
com.google.protobuf.MessageLiteOrBuilder,com.google.protobuf.MessageOrBuilder
- All Known Implementing Classes:
GPUOptions.Experimental.VirtualDevices,GPUOptions.Experimental.VirtualDevices.Builder
- Enclosing class:
- GPUOptions.Experimental
public static interface GPUOptions.Experimental.VirtualDevicesOrBuilder
extends com.google.protobuf.MessageOrBuilder
-
Method Summary
Modifier and Type Method Description floatgetMemoryLimitMb(int index)Per "virtual" device memory limit, in MB.intgetMemoryLimitMbCount()Per "virtual" device memory limit, in MB.java.util.List<java.lang.Float>getMemoryLimitMbList()Per "virtual" device memory limit, in MB.Methods inherited from interface com.google.protobuf.MessageOrBuilder
findInitializationErrors, getAllFields, getDefaultInstanceForType, getDescriptorForType, getField, getInitializationErrorString, getOneofFieldDescriptor, getRepeatedField, getRepeatedFieldCount, getUnknownFields, hasField, hasOneof
-
Method Details
-
getMemoryLimitMbList
java.util.List<java.lang.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;- Returns:
- A list containing the memoryLimitMb.
-
getMemoryLimitMbCount
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;- Returns:
- The count of memoryLimitMb.
-
getMemoryLimitMb
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;- Parameters:
index- The index of the element to return.- Returns:
- The memoryLimitMb at the given index.
-