Caffe2 - Python API
A deep learning, cross platform ML framework
Public Member Functions | Public Attributes | List of all members
common_nn.CriterionTest Class Reference
Inheritance diagram for common_nn.CriterionTest:
common_nn.TestBase test_nn.NewCriterionTest

Public Member Functions

def __init__ (self, args, kwargs)
 
def __call__ (self, test_case)
 
def test_cuda (self, test_case)
 
- Public Member Functions inherited from common_nn.TestBase
def __init__ (self, constructor, desc='', reference_fn=None, fullname=None, kwargs)
 
def get_name (self)
 
def constructor_args (self)
 
def extra_args (self)
 
def __call__ (self, test_case)
 

Public Attributes

 should_test_cuda
 
 check_forward_only
 
- Public Attributes inherited from common_nn.TestBase
 desc
 
 fullname
 
 constructor
 
 reference_fn
 

Detailed Description

Definition at line 3314 of file common_nn.py.


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