Package org.tensorflow.framework

Class GPUOptions.Experimental.VirtualDevices.Builder

    • Method Detail

      • getDescriptor

        public static final org.nd4j.shade.protobuf.Descriptors.Descriptor getDescriptor()

      • internalGetFieldAccessorTable

        protected org.nd4j.shade.protobuf.GeneratedMessageV3.FieldAccessorTable internalGetFieldAccessorTable()
        Specified by:
        internalGetFieldAccessorTable in class org.nd4j.shade.protobuf.GeneratedMessageV3.Builder<GPUOptions.Experimental.VirtualDevices.Builder>

      • getDescriptorForType

        public org.nd4j.shade.protobuf.Descriptors.Descriptor getDescriptorForType()
        Specified by:
        getDescriptorForType in interface org.nd4j.shade.protobuf.Message.Builder
        Specified by:
        getDescriptorForType in interface org.nd4j.shade.protobuf.MessageOrBuilder
        Overrides:
        getDescriptorForType in class org.nd4j.shade.protobuf.GeneratedMessageV3.Builder<GPUOptions.Experimental.VirtualDevices.Builder>

      • getDefaultInstanceForType

        public GPUOptions.Experimental.VirtualDevices getDefaultInstanceForType()
        Specified by:
        getDefaultInstanceForType in interface org.nd4j.shade.protobuf.MessageLiteOrBuilder
        Specified by:
        getDefaultInstanceForType in interface org.nd4j.shade.protobuf.MessageOrBuilder

      • build

        public GPUOptions.Experimental.VirtualDevices build()
        Specified by:
        build in interface org.nd4j.shade.protobuf.Message.Builder
        Specified by:
        build in interface org.nd4j.shade.protobuf.MessageLite.Builder

      • buildPartial

        public GPUOptions.Experimental.VirtualDevices buildPartial()
        Specified by:
        buildPartial in interface org.nd4j.shade.protobuf.Message.Builder
        Specified by:
        buildPartial in interface org.nd4j.shade.protobuf.MessageLite.Builder

      • isInitialized

        public final boolean isInitialized()
        Specified by:
        isInitialized in interface org.nd4j.shade.protobuf.MessageLiteOrBuilder
        Overrides:
        isInitialized in class org.nd4j.shade.protobuf.GeneratedMessageV3.Builder<GPUOptions.Experimental.VirtualDevices.Builder>

      • getMemoryLimitMbList

        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;
        Specified by:
        getMemoryLimitMbList in interface GPUOptions.Experimental.VirtualDevicesOrBuilder
        Returns:
        A list containing the memoryLimitMb.

      • getMemoryLimitMbCount

        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;
        Specified by:
        getMemoryLimitMbCount in interface GPUOptions.Experimental.VirtualDevicesOrBuilder
        Returns:
        The count of memoryLimitMb.

      • getMemoryLimitMb

        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;
        Specified by:
        getMemoryLimitMb in interface GPUOptions.Experimental.VirtualDevicesOrBuilder
        Parameters:
        index - The index of the element to return.
        Returns:
        The memoryLimitMb at the given index.

      • setMemoryLimitMb

        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;
        Parameters:
        index - The index to set the value at.
        value - The memoryLimitMb to set.
        Returns:
        This builder for chaining.

      • addMemoryLimitMb

        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;
        Parameters:
        value - The memoryLimitMb to add.
        Returns:
        This builder for chaining.

      • addAllMemoryLimitMb

        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;
        Parameters:
        values - The memoryLimitMb to add.
        Returns:
        This builder for chaining.

      • clearMemoryLimitMb

        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;
        Returns:
        This builder for chaining.