torch.cuda.streams.Stream Class Reference

Inheritance diagram for torch.cuda.streams.Stream:

Public Member Functions
def	__new__ (cls, device=None, priority=0, kwargs)
def	wait_event (self, event)
def	wait_stream (self, stream)
def	record_event (self, event=None)
def	query (self)
def	synchronize (self)
def	__eq__ (self, o)
def	__hash__ (self)
def	__repr__ (self)

Detailed Description

Definition at line 5 of file streams.py.

---

The documentation for this class was generated from the following file:

• torch/cuda/streams.py