public final class GPUOptions extends com.google.protobuf.GeneratedMessageV3 implements GPUOptionsOrBuilder
tensorflow.GPUOptions
Modifier and Type | Class and Description |
---|---|
static class |
GPUOptions.Builder
Protobuf type
tensorflow.GPUOptions |
com.google.protobuf.GeneratedMessageV3.BuilderParent, com.google.protobuf.GeneratedMessageV3.ExtendableBuilder<MessageType extends com.google.protobuf.GeneratedMessageV3.ExtendableMessage,BuilderType extends com.google.protobuf.GeneratedMessageV3.ExtendableBuilder<MessageType,BuilderType>>, com.google.protobuf.GeneratedMessageV3.ExtendableMessage<MessageType extends com.google.protobuf.GeneratedMessageV3.ExtendableMessage>, com.google.protobuf.GeneratedMessageV3.ExtendableMessageOrBuilder<MessageType extends com.google.protobuf.GeneratedMessageV3.ExtendableMessage>, com.google.protobuf.GeneratedMessageV3.FieldAccessorTable
Modifier and Type | Field and Description |
---|---|
static int |
ALLOCATOR_TYPE_FIELD_NUMBER |
static int |
ALLOW_GROWTH_FIELD_NUMBER |
static int |
DEFERRED_DELETION_BYTES_FIELD_NUMBER |
static int |
FORCE_GPU_COMPATIBLE_FIELD_NUMBER |
static int |
PER_PROCESS_GPU_MEMORY_FRACTION_FIELD_NUMBER |
static int |
POLLING_ACTIVE_DELAY_USECS_FIELD_NUMBER |
static int |
POLLING_INACTIVE_DELAY_MSECS_FIELD_NUMBER |
static int |
VISIBLE_DEVICE_LIST_FIELD_NUMBER |
Modifier and Type | Method and Description |
---|---|
boolean |
equals(Object obj) |
String |
getAllocatorType()
The type of GPU allocation strategy to use.
|
com.google.protobuf.ByteString |
getAllocatorTypeBytes()
The type of GPU allocation strategy to use.
|
boolean |
getAllowGrowth()
If true, the allocator does not pre-allocate the entire specified
GPU memory region, instead starting small and growing as needed.
|
static GPUOptions |
getDefaultInstance() |
GPUOptions |
getDefaultInstanceForType() |
long |
getDeferredDeletionBytes()
Delay deletion of up to this many bytes to reduce the number of
interactions with gpu driver code.
|
static com.google.protobuf.Descriptors.Descriptor |
getDescriptor() |
boolean |
getForceGpuCompatible()
Force all tensors to be gpu_compatible.
|
com.google.protobuf.Parser<GPUOptions> |
getParserForType() |
double |
getPerProcessGpuMemoryFraction()
A value between 0 and 1 that indicates what fraction of the
available GPU memory to pre-allocate for each process.
|
int |
getPollingActiveDelayUsecs()
In the event polling loop sleep this many microseconds between
PollEvents calls, when the queue is not empty.
|
int |
getPollingInactiveDelayMsecs()
In the event polling loop sleep this many millisconds between
PollEvents calls, when the queue is empty.
|
int |
getSerializedSize() |
com.google.protobuf.UnknownFieldSet |
getUnknownFields() |
String |
getVisibleDeviceList()
A comma-separated list of GPU ids that determines the 'visible'
to 'virtual' mapping of GPU devices.
|
com.google.protobuf.ByteString |
getVisibleDeviceListBytes()
A comma-separated list of GPU ids that determines the 'visible'
to 'virtual' mapping of GPU devices.
|
int |
hashCode() |
protected com.google.protobuf.GeneratedMessageV3.FieldAccessorTable |
internalGetFieldAccessorTable() |
boolean |
isInitialized() |
static GPUOptions.Builder |
newBuilder() |
static GPUOptions.Builder |
newBuilder(GPUOptions prototype) |
GPUOptions.Builder |
newBuilderForType() |
protected GPUOptions.Builder |
newBuilderForType(com.google.protobuf.GeneratedMessageV3.BuilderParent parent) |
static GPUOptions |
parseDelimitedFrom(InputStream input) |
static GPUOptions |
parseDelimitedFrom(InputStream input,
com.google.protobuf.ExtensionRegistryLite extensionRegistry) |
static GPUOptions |
parseFrom(byte[] data) |
static GPUOptions |
parseFrom(byte[] data,
com.google.protobuf.ExtensionRegistryLite extensionRegistry) |
static GPUOptions |
parseFrom(ByteBuffer data) |
static GPUOptions |
parseFrom(ByteBuffer data,
com.google.protobuf.ExtensionRegistryLite extensionRegistry) |
static GPUOptions |
parseFrom(com.google.protobuf.ByteString data) |
static GPUOptions |
parseFrom(com.google.protobuf.ByteString data,
com.google.protobuf.ExtensionRegistryLite extensionRegistry) |
static GPUOptions |
parseFrom(com.google.protobuf.CodedInputStream input) |
static GPUOptions |
parseFrom(com.google.protobuf.CodedInputStream input,
com.google.protobuf.ExtensionRegistryLite extensionRegistry) |
static GPUOptions |
parseFrom(InputStream input) |
static GPUOptions |
parseFrom(InputStream input,
com.google.protobuf.ExtensionRegistryLite extensionRegistry) |
static com.google.protobuf.Parser<GPUOptions> |
parser() |
GPUOptions.Builder |
toBuilder() |
void |
writeTo(com.google.protobuf.CodedOutputStream output) |
computeStringSize, computeStringSizeNoTag, getAllFields, getDescriptorForType, getField, getOneofFieldDescriptor, getRepeatedField, getRepeatedFieldCount, hasField, hasOneof, internalGetMapField, makeExtensionsImmutable, newBuilderForType, parseDelimitedWithIOException, parseDelimitedWithIOException, parseUnknownField, parseWithIOException, parseWithIOException, parseWithIOException, parseWithIOException, serializeBooleanMapTo, serializeIntegerMapTo, serializeLongMapTo, serializeStringMapTo, writeReplace, writeString, writeStringNoTag
findInitializationErrors, getInitializationErrorString, hashBoolean, hashEnum, hashEnumList, hashFields, hashLong, toString
addAll, checkByteStringIsUtf8, toByteArray, toByteString, writeDelimitedTo, writeTo
clone, finalize, getClass, notify, notifyAll, wait, wait, wait
public static final int PER_PROCESS_GPU_MEMORY_FRACTION_FIELD_NUMBER
public static final int ALLOCATOR_TYPE_FIELD_NUMBER
public static final int DEFERRED_DELETION_BYTES_FIELD_NUMBER
public static final int ALLOW_GROWTH_FIELD_NUMBER
public static final int VISIBLE_DEVICE_LIST_FIELD_NUMBER
public static final int POLLING_ACTIVE_DELAY_USECS_FIELD_NUMBER
public static final int POLLING_INACTIVE_DELAY_MSECS_FIELD_NUMBER
public static final int FORCE_GPU_COMPATIBLE_FIELD_NUMBER
public final com.google.protobuf.UnknownFieldSet getUnknownFields()
getUnknownFields
in interface com.google.protobuf.MessageOrBuilder
getUnknownFields
in class com.google.protobuf.GeneratedMessageV3
public static final com.google.protobuf.Descriptors.Descriptor getDescriptor()
protected com.google.protobuf.GeneratedMessageV3.FieldAccessorTable internalGetFieldAccessorTable()
internalGetFieldAccessorTable
in class com.google.protobuf.GeneratedMessageV3
public double getPerProcessGpuMemoryFraction()
A value between 0 and 1 that indicates what fraction of the available GPU memory to pre-allocate for each process. 1 means to pre-allocate all of the GPU memory, 0.5 means the process allocates ~50% of the available GPU memory.
double per_process_gpu_memory_fraction = 1;
getPerProcessGpuMemoryFraction
in interface GPUOptionsOrBuilder
public String getAllocatorType()
The type of GPU allocation strategy to use. Allowed values: "": The empty string (default) uses a system-chosen default which may change over time. "BFC": A "Best-fit with coalescing" algorithm, simplified from a version of dlmalloc.
string allocator_type = 2;
getAllocatorType
in interface GPUOptionsOrBuilder
public com.google.protobuf.ByteString getAllocatorTypeBytes()
The type of GPU allocation strategy to use. Allowed values: "": The empty string (default) uses a system-chosen default which may change over time. "BFC": A "Best-fit with coalescing" algorithm, simplified from a version of dlmalloc.
string allocator_type = 2;
getAllocatorTypeBytes
in interface GPUOptionsOrBuilder
public long getDeferredDeletionBytes()
Delay deletion of up to this many bytes to reduce the number of interactions with gpu driver code. If 0, the system chooses a reasonable default (several MBs).
int64 deferred_deletion_bytes = 3;
getDeferredDeletionBytes
in interface GPUOptionsOrBuilder
public boolean getAllowGrowth()
If true, the allocator does not pre-allocate the entire specified GPU memory region, instead starting small and growing as needed.
bool allow_growth = 4;
getAllowGrowth
in interface GPUOptionsOrBuilder
public String getVisibleDeviceList()
A comma-separated list of GPU ids that determines the 'visible' to 'virtual' mapping of GPU devices. For example, if TensorFlow can see 8 GPU devices in the process, and one wanted to map visible GPU devices 5 and 3 as "/device:GPU:0", and "/device:GPU:1", then one would specify this field as "5,3". This field is similar in spirit to the CUDA_VISIBLE_DEVICES environment variable, except it applies to the visible GPU devices in the process. NOTE: The GPU driver provides the process with the visible GPUs in an order which is not guaranteed to have any correlation to the *physical* GPU id in the machine. This field is used for remapping "visible" to "virtual", which means this operates only after the process starts. Users are required to use vendor specific mechanisms (e.g., CUDA_VISIBLE_DEVICES) to control the physical to visible device mapping prior to invoking TensorFlow.
string visible_device_list = 5;
getVisibleDeviceList
in interface GPUOptionsOrBuilder
public com.google.protobuf.ByteString getVisibleDeviceListBytes()
A comma-separated list of GPU ids that determines the 'visible' to 'virtual' mapping of GPU devices. For example, if TensorFlow can see 8 GPU devices in the process, and one wanted to map visible GPU devices 5 and 3 as "/device:GPU:0", and "/device:GPU:1", then one would specify this field as "5,3". This field is similar in spirit to the CUDA_VISIBLE_DEVICES environment variable, except it applies to the visible GPU devices in the process. NOTE: The GPU driver provides the process with the visible GPUs in an order which is not guaranteed to have any correlation to the *physical* GPU id in the machine. This field is used for remapping "visible" to "virtual", which means this operates only after the process starts. Users are required to use vendor specific mechanisms (e.g., CUDA_VISIBLE_DEVICES) to control the physical to visible device mapping prior to invoking TensorFlow.
string visible_device_list = 5;
getVisibleDeviceListBytes
in interface GPUOptionsOrBuilder
public int getPollingActiveDelayUsecs()
In the event polling loop sleep this many microseconds between PollEvents calls, when the queue is not empty. If value is not set or set to 0, gets set to a non-zero default.
int32 polling_active_delay_usecs = 6;
getPollingActiveDelayUsecs
in interface GPUOptionsOrBuilder
public int getPollingInactiveDelayMsecs()
In the event polling loop sleep this many millisconds between PollEvents calls, when the queue is empty. If value is not set or set to 0, gets set to a non-zero default.
int32 polling_inactive_delay_msecs = 7;
getPollingInactiveDelayMsecs
in interface GPUOptionsOrBuilder
public boolean getForceGpuCompatible()
Force all tensors to be gpu_compatible. On a GPU-enabled TensorFlow, enabling this option forces all CPU tensors to be allocated with Cuda pinned memory. Normally, TensorFlow will infer which tensors should be allocated as the pinned memory. But in case where the inference is incomplete, this option can significantly speed up the cross-device memory copy performance as long as it fits the memory. Note that this option is not something that should be enabled by default for unknown or very large models, since all Cuda pinned memory is unpageable, having too much pinned memory might negatively impact the overall host system performance.
bool force_gpu_compatible = 8;
getForceGpuCompatible
in interface GPUOptionsOrBuilder
public final boolean isInitialized()
isInitialized
in interface com.google.protobuf.MessageLiteOrBuilder
isInitialized
in class com.google.protobuf.GeneratedMessageV3
public void writeTo(com.google.protobuf.CodedOutputStream output) throws IOException
writeTo
in interface com.google.protobuf.MessageLite
writeTo
in class com.google.protobuf.GeneratedMessageV3
IOException
public int getSerializedSize()
getSerializedSize
in interface com.google.protobuf.MessageLite
getSerializedSize
in class com.google.protobuf.GeneratedMessageV3
public boolean equals(Object obj)
equals
in interface com.google.protobuf.Message
equals
in class com.google.protobuf.AbstractMessage
public int hashCode()
hashCode
in interface com.google.protobuf.Message
hashCode
in class com.google.protobuf.AbstractMessage
public static GPUOptions parseFrom(ByteBuffer data) throws com.google.protobuf.InvalidProtocolBufferException
com.google.protobuf.InvalidProtocolBufferException
public static GPUOptions parseFrom(ByteBuffer data, com.google.protobuf.ExtensionRegistryLite extensionRegistry) throws com.google.protobuf.InvalidProtocolBufferException
com.google.protobuf.InvalidProtocolBufferException
public static GPUOptions parseFrom(com.google.protobuf.ByteString data) throws com.google.protobuf.InvalidProtocolBufferException
com.google.protobuf.InvalidProtocolBufferException
public static GPUOptions parseFrom(com.google.protobuf.ByteString data, com.google.protobuf.ExtensionRegistryLite extensionRegistry) throws com.google.protobuf.InvalidProtocolBufferException
com.google.protobuf.InvalidProtocolBufferException
public static GPUOptions parseFrom(byte[] data) throws com.google.protobuf.InvalidProtocolBufferException
com.google.protobuf.InvalidProtocolBufferException
public static GPUOptions parseFrom(byte[] data, com.google.protobuf.ExtensionRegistryLite extensionRegistry) throws com.google.protobuf.InvalidProtocolBufferException
com.google.protobuf.InvalidProtocolBufferException
public static GPUOptions parseFrom(InputStream input) throws IOException
IOException
public static GPUOptions parseFrom(InputStream input, com.google.protobuf.ExtensionRegistryLite extensionRegistry) throws IOException
IOException
public static GPUOptions parseDelimitedFrom(InputStream input) throws IOException
IOException
public static GPUOptions parseDelimitedFrom(InputStream input, com.google.protobuf.ExtensionRegistryLite extensionRegistry) throws IOException
IOException
public static GPUOptions parseFrom(com.google.protobuf.CodedInputStream input) throws IOException
IOException
public static GPUOptions parseFrom(com.google.protobuf.CodedInputStream input, com.google.protobuf.ExtensionRegistryLite extensionRegistry) throws IOException
IOException
public GPUOptions.Builder newBuilderForType()
newBuilderForType
in interface com.google.protobuf.Message
newBuilderForType
in interface com.google.protobuf.MessageLite
public static GPUOptions.Builder newBuilder()
public static GPUOptions.Builder newBuilder(GPUOptions prototype)
public GPUOptions.Builder toBuilder()
toBuilder
in interface com.google.protobuf.Message
toBuilder
in interface com.google.protobuf.MessageLite
protected GPUOptions.Builder newBuilderForType(com.google.protobuf.GeneratedMessageV3.BuilderParent parent)
newBuilderForType
in class com.google.protobuf.GeneratedMessageV3
public static GPUOptions getDefaultInstance()
public static com.google.protobuf.Parser<GPUOptions> parser()
public com.google.protobuf.Parser<GPUOptions> getParserForType()
getParserForType
in interface com.google.protobuf.Message
getParserForType
in interface com.google.protobuf.MessageLite
getParserForType
in class com.google.protobuf.GeneratedMessageV3
public GPUOptions getDefaultInstanceForType()
getDefaultInstanceForType
in interface com.google.protobuf.MessageLiteOrBuilder
getDefaultInstanceForType
in interface com.google.protobuf.MessageOrBuilder
Copyright © 2015–2017. All rights reserved.