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
    float getMemoryLimitMb​(int index)
    Per "virtual" device memory limit, in MB.
    int getMemoryLimitMbCount()
    Per "virtual" device memory limit, in MB.
    java.util.List<java.lang.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.
    java.util.List<java.lang.Integer> getPriorityList()
    Priority values to use with the virtual devices.

    Methods inherited from interface com.google.protobuf.MessageLiteOrBuilder

    isInitialized

    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.

    • getPriorityList

      java.util.List<java.lang.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;
      Returns:
      A list containing the priority.

    • getPriorityCount

      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;
      Returns:
      The count of priority.

    • getPriority

      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;
      Parameters:
      index - The index of the element to return.
      Returns:
      The priority at the given index.