Caffe2 - Python API
A deep learning, cross platform ML framework
Public Member Functions | Public Attributes | List of all members
torch.cuda.device Class Reference
Inheritance diagram for torch.cuda.device:
torch.cuda.device_of

Public Member Functions

def __init__ (self, device)
 
def __enter__ (self)
 
def __exit__ (self, args)
 

Public Attributes

 idx
 
 prev_idx
 

Detailed Description

Definition at line 212 of file __init__.py.


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