Uses of Class
org.bytedeco.pytorch.Stream

Packages that use Stream

Package	Description
org.bytedeco.pytorch
org.bytedeco.pytorch.cuda
org.bytedeco.pytorch.global

Uses of Stream in org.bytedeco.pytorch

Methods in org.bytedeco.pytorch that return Stream

Modifier and Type	Method and Description
Stream	DeviceGuardImplInterface.exchangeStream(Stream arg0) Set a stream to be the thread local current stream for its device.
Stream	StreamOptional.get()
Stream	DeviceGuardImplInterface.getDefaultStream(Device arg0) Get the default stream for a given device.
Stream	DeviceGuardImplInterface.getStream(Device arg0) Get the current stream for a given device.
Stream	DeviceGuardImplInterface.getStreamFromGlobalPool(Device arg0)
Stream	DeviceGuardImplInterface.getStreamFromGlobalPool(Device arg0, boolean isHighPriority) Get a stream from the global pool for a given device.
Stream	IValue.toStream()
static Stream	Stream.unpack3(long stream_id, byte device_index, byte device_type)
static Stream	Stream.unpack3(long stream_id, byte device_index, torch.DeviceType device_type)

Methods in org.bytedeco.pytorch with parameters of type Stream

Modifier and Type	Method and Description
void	DeviceGuardImplInterface.block(Pointer arg0, Stream arg1) Does nothing if the event has not been scheduled to be recorded.
boolean	Stream.equals(Stream other)
Stream	DeviceGuardImplInterface.exchangeStream(Stream arg0) Set a stream to be the thread local current stream for its device.
boolean	Stream.notEquals(Stream other)
StreamOptional	StreamOptional.put(Stream value)
boolean	DeviceGuardImplInterface.queryStream(Stream arg0) Return true if all the work previously enqueued on the stream for asynchronous execution has completed running on the device.
void	Tensor.record_stream(Stream s)
void	DeviceGuardImplInterface.record(PointerPointer arg0, Stream arg1, byte arg2, torch.EventFlag arg3) Increments the event's version and enqueues a job with this version in the stream's work queue.
void	DeviceGuardImplInterface.record(Pointer arg0, Stream arg1, byte arg2, int arg3)
void	DeviceGuardImplInterface.record(Pointer arg0, Stream arg1, byte arg2, torch.EventFlag arg3)
void	DeviceGuardImplInterface.recordDataPtrOnStream(DataPtr arg0, Stream arg1) Ensure the caching allocator (if any) is aware that the given DataPtr is being used on the given stream, and that it should thus avoid recycling the DataPtr until all work on that stream is done.
void	DeviceGuardImplInterface.synchronizeStream(Stream arg0) Wait (by blocking the calling thread) until all the work previously enqueued on the stream has completed running on the device.

Constructors in org.bytedeco.pytorch with parameters of type Stream

Constructor and Description
IValue(Stream s)
StreamOptional(Stream value)

Uses of Stream in org.bytedeco.pytorch.cuda

Methods in org.bytedeco.pytorch.cuda that return Stream

Modifier and Type	Method and Description
Stream	CUDAStream.asStream() Implicit conversion to Stream (a.k.a., forget that the stream is a CUDA stream).
Stream	CUDAStream.unwrap() Explicit conversion to Stream.

Methods in org.bytedeco.pytorch.cuda with parameters of type Stream

Modifier and Type	Method and Description
void	CUDAStreamGuard.reset_stream(Stream stream) Resets the currently set stream to the original stream and the currently set device to the original device.

Constructors in org.bytedeco.pytorch.cuda with parameters of type Stream

Constructor and Description
CUDAStream(CUDAStream.Unchecked arg0, Stream stream) Construct a CUDAStream from a Stream with no error checking.
CUDAStream(int arg0, Stream stream)
CUDAStream(Stream stream) Construct a CUDAStream from a Stream.
CUDAStreamGuard(Stream stream) Set the current CUDA device to the device associated with the passed stream, and set the current CUDA stream on that device to the passed stream.

Uses of Stream in org.bytedeco.pytorch.global

Methods in org.bytedeco.pytorch.global with parameters of type Stream

Modifier and Type	Method and Description
static Pointer	torch.shiftLeft(Pointer stream, Stream s) An index representing a specific stream.