Package org.tensorflow.framework

Interface GPUOptions.Experimental.StreamMergeOptionsOrBuilder

All Superinterfaces:
com.google.protobuf.MessageLiteOrBuilder, com.google.protobuf.MessageOrBuilder
All Known Implementing Classes:
GPUOptions.Experimental.StreamMergeOptions, GPUOptions.Experimental.StreamMergeOptions.Builder
Enclosing class:
GPUOptions.Experimental
public static interface GPUOptions.Experimental.StreamMergeOptionsOrBuilder extends com.google.protobuf.MessageOrBuilder

  • Method Summary

    Modifier and Type
    Method
    Description
    boolean
    getMergeDeviceToDeviceStream()
    If true, the compute stream will be used for device_to_device copy as well.
    boolean
    getMergeDeviceToHostStream()
    If true, the compute stream will be used for device_to_host copy as well.
    boolean
    getMergeHostToDeviceStream()
    If true, the compute stream will be used for host_to_device copy as well.

    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

    • getMergeHostToDeviceStream

      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;
      Returns:
      The mergeHostToDeviceStream.

    • getMergeDeviceToHostStream

      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;
      Returns:
      The mergeDeviceToHostStream.

    • getMergeDeviceToDeviceStream

      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;
      Returns:
      The mergeDeviceToDeviceStream.