Caffe2 - Python API
A deep learning, cross platform ML framework
Public Member Functions | Static Public Member Functions | Public Attributes | List of all members
common_utils.CudaMemoryLeakCheck Class Reference

Public Member Functions

def __init__ (self, testcase, name=None)
 
def __enter__ (self)
 
def __exit__ (self, exec_type, exec_value, traceback)
 

Static Public Member Functions

def get_cuda_memory_usage ()
 

Public Attributes

 name
 
 testcase
 
 befores
 

Detailed Description

Definition at line 226 of file common_utils.py.


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