torch.autograd.profiler.EnforceUnique Class Reference

CUDA checkpoints. More...

Inheritance diagram for torch.autograd.profiler.EnforceUnique:

Public Member Functions
def	__init__ (self)
def	see (self, key)

Public Attributes
	seen

Detailed Description

CUDA checkpoints.

Raises an error if a key is seen more than once.

Definition at line 497 of file profiler.py.

---

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

• torch/autograd/profiler.py