Package org.tensorflow.framework

Class GPUOptions.Experimental.StreamMergeOptions

java.lang.Object
com.google.protobuf.AbstractMessageLite
com.google.protobuf.AbstractMessage
com.google.protobuf.GeneratedMessageV3
org.tensorflow.framework.GPUOptions.Experimental.StreamMergeOptions
All Implemented Interfaces:
com.google.protobuf.Message, com.google.protobuf.MessageLite, com.google.protobuf.MessageLiteOrBuilder, com.google.protobuf.MessageOrBuilder, Serializable, GPUOptions.Experimental.StreamMergeOptionsOrBuilder
Enclosing class:
GPUOptions.Experimental
public static final class GPUOptions.Experimental.StreamMergeOptions extends com.google.protobuf.GeneratedMessageV3 implements GPUOptions.Experimental.StreamMergeOptionsOrBuilder
 Whether to merge data transfer streams into the compute stream in the
 same stream group. Stream merging helps reduce the overhead caused by
 stream synchronization, especially when data transfers are frequent. For
 example, setting "merge_host_to_device_stream = true" will make the
 compute stream responsible for both computation and host to device memory
 copy.
Protobuf type tensorflow.GPUOptions.Experimental.StreamMergeOptions
See Also:

  • Field Details

    • MERGE_HOST_TO_DEVICE_STREAM_FIELD_NUMBER

      public static final int MERGE_HOST_TO_DEVICE_STREAM_FIELD_NUMBER
      See Also:

    • MERGE_DEVICE_TO_HOST_STREAM_FIELD_NUMBER

      public static final int MERGE_DEVICE_TO_HOST_STREAM_FIELD_NUMBER
      See Also:

    • MERGE_DEVICE_TO_DEVICE_STREAM_FIELD_NUMBER

      public static final int MERGE_DEVICE_TO_DEVICE_STREAM_FIELD_NUMBER
      See Also:

  • Method Details

    • newInstance

      protected Object newInstance(com.google.protobuf.GeneratedMessageV3.UnusedPrivateParameter unused)
      Overrides:
      newInstance in class com.google.protobuf.GeneratedMessageV3

    • getDescriptor

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

    • internalGetFieldAccessorTable

      protected com.google.protobuf.GeneratedMessageV3.FieldAccessorTable internalGetFieldAccessorTable()
      Specified by:
      internalGetFieldAccessorTable in class com.google.protobuf.GeneratedMessageV3

    • getMergeHostToDeviceStream

      public boolean getMergeHostToDeviceStream()
       If true, the compute stream will be used for host_to_device copy as
 well. It's no longer necessary to record an event before the copy to
 let the copy stream wait for the compute stream to finish. There is
 also no need to wait for the copy to complete before executing the
 callback function.
      bool merge_host_to_device_stream = 1;
      Specified by:
      getMergeHostToDeviceStream in interface GPUOptions.Experimental.StreamMergeOptionsOrBuilder
      Returns:
      The mergeHostToDeviceStream.

    • getMergeDeviceToHostStream

      public boolean getMergeDeviceToHostStream()
       If true, the compute stream will be used for device_to_host copy as
 well. It's no longer necessary to record an event before the copy to
 let the copy stream wait for the compute stream to finish.
      bool merge_device_to_host_stream = 2;
      Specified by:
      getMergeDeviceToHostStream in interface GPUOptions.Experimental.StreamMergeOptionsOrBuilder
      Returns:
      The mergeDeviceToHostStream.

    • getMergeDeviceToDeviceStream

      public boolean getMergeDeviceToDeviceStream()
       If true, the compute stream will be used for device_to_device copy as
 well. It's no longer necessary to record an event before the copy to
 let the copy stream wait for the compute stream of the sending device
 to finish. There is also no need to wait for the compute stream of the
 receiving device to finish if the copy is within the same device.
      bool merge_device_to_device_stream = 3;
      Specified by:
      getMergeDeviceToDeviceStream in interface GPUOptions.Experimental.StreamMergeOptionsOrBuilder
      Returns:
      The mergeDeviceToDeviceStream.

    • isInitialized

      public final boolean isInitialized()
      Specified by:
      isInitialized in interface com.google.protobuf.MessageLiteOrBuilder
      Overrides:
      isInitialized in class com.google.protobuf.GeneratedMessageV3

    • writeTo

      public void writeTo(com.google.protobuf.CodedOutputStream output) throws IOException
      Specified by:
      writeTo in interface com.google.protobuf.MessageLite
      Overrides:
      writeTo in class com.google.protobuf.GeneratedMessageV3
      Throws:
      IOException

    • getSerializedSize

      public int getSerializedSize()
      Specified by:
      getSerializedSize in interface com.google.protobuf.MessageLite
      Overrides:
      getSerializedSize in class com.google.protobuf.GeneratedMessageV3

    • equals

      public boolean equals(Object obj)
      Specified by:
      equals in interface com.google.protobuf.Message
      Overrides:
      equals in class com.google.protobuf.AbstractMessage

    • hashCode

      public int hashCode()
      Specified by:
      hashCode in interface com.google.protobuf.Message
      Overrides:
      hashCode in class com.google.protobuf.AbstractMessage

    • parseFrom

      public static GPUOptions.Experimental.StreamMergeOptions parseFrom(ByteBuffer data) throws com.google.protobuf.InvalidProtocolBufferException
      Throws:
      com.google.protobuf.InvalidProtocolBufferException

    • parseFrom

      public static GPUOptions.Experimental.StreamMergeOptions parseFrom(ByteBuffer data, com.google.protobuf.ExtensionRegistryLite extensionRegistry) throws com.google.protobuf.InvalidProtocolBufferException
      Throws:
      com.google.protobuf.InvalidProtocolBufferException

    • parseFrom

      public static GPUOptions.Experimental.StreamMergeOptions parseFrom(com.google.protobuf.ByteString data) throws com.google.protobuf.InvalidProtocolBufferException
      Throws:
      com.google.protobuf.InvalidProtocolBufferException

    • parseFrom

      public static GPUOptions.Experimental.StreamMergeOptions parseFrom(com.google.protobuf.ByteString data, com.google.protobuf.ExtensionRegistryLite extensionRegistry) throws com.google.protobuf.InvalidProtocolBufferException
      Throws:
      com.google.protobuf.InvalidProtocolBufferException

    • parseFrom

      public static GPUOptions.Experimental.StreamMergeOptions parseFrom(byte[] data) throws com.google.protobuf.InvalidProtocolBufferException
      Throws:
      com.google.protobuf.InvalidProtocolBufferException

    • parseFrom

      public static GPUOptions.Experimental.StreamMergeOptions parseFrom(byte[] data, com.google.protobuf.ExtensionRegistryLite extensionRegistry) throws com.google.protobuf.InvalidProtocolBufferException
      Throws:
      com.google.protobuf.InvalidProtocolBufferException

    • parseFrom

      public static GPUOptions.Experimental.StreamMergeOptions parseFrom(InputStream input) throws IOException
      Throws:
      IOException

    • parseFrom

      public static GPUOptions.Experimental.StreamMergeOptions parseFrom(InputStream input, com.google.protobuf.ExtensionRegistryLite extensionRegistry) throws IOException
      Throws:
      IOException

    • parseDelimitedFrom

      public static GPUOptions.Experimental.StreamMergeOptions parseDelimitedFrom(InputStream input) throws IOException
      Throws:
      IOException

    • parseDelimitedFrom

      public static GPUOptions.Experimental.StreamMergeOptions parseDelimitedFrom(InputStream input, com.google.protobuf.ExtensionRegistryLite extensionRegistry) throws IOException
      Throws:
      IOException

    • parseFrom

      public static GPUOptions.Experimental.StreamMergeOptions parseFrom(com.google.protobuf.CodedInputStream input) throws IOException
      Throws:
      IOException

    • parseFrom

      public static GPUOptions.Experimental.StreamMergeOptions parseFrom(com.google.protobuf.CodedInputStream input, com.google.protobuf.ExtensionRegistryLite extensionRegistry) throws IOException
      Throws:
      IOException

    • newBuilderForType

      public GPUOptions.Experimental.StreamMergeOptions.Builder newBuilderForType()
      Specified by:
      newBuilderForType in interface com.google.protobuf.Message
      Specified by:
      newBuilderForType in interface com.google.protobuf.MessageLite

    • newBuilder

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

    • newBuilder

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

    • toBuilder

      public GPUOptions.Experimental.StreamMergeOptions.Builder toBuilder()
      Specified by:
      toBuilder in interface com.google.protobuf.Message
      Specified by:
      toBuilder in interface com.google.protobuf.MessageLite

    • newBuilderForType

      protected GPUOptions.Experimental.StreamMergeOptions.Builder newBuilderForType(com.google.protobuf.GeneratedMessageV3.BuilderParent parent)
      Specified by:
      newBuilderForType in class com.google.protobuf.GeneratedMessageV3

    • getDefaultInstance

      public static GPUOptions.Experimental.StreamMergeOptions getDefaultInstance()

    • parser

      public static com.google.protobuf.Parser<GPUOptions.Experimental.StreamMergeOptions> parser()

    • getParserForType

      public com.google.protobuf.Parser<GPUOptions.Experimental.StreamMergeOptions> getParserForType()
      Specified by:
      getParserForType in interface com.google.protobuf.Message
      Specified by:
      getParserForType in interface com.google.protobuf.MessageLite
      Overrides:
      getParserForType in class com.google.protobuf.GeneratedMessageV3

    • getDefaultInstanceForType

      public GPUOptions.Experimental.StreamMergeOptions getDefaultInstanceForType()
      Specified by:
      getDefaultInstanceForType in interface com.google.protobuf.MessageLiteOrBuilder
      Specified by:
      getDefaultInstanceForType in interface com.google.protobuf.MessageOrBuilder