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

Public Member Functions

def __init__ (self, args, kwargs)
 
def __call__ (self, test_case)
 
def noncontiguize (self, obj)
 
def test_noncontig (self, test_case, module, input)
 
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

 jacobian_input
 
 should_test_cuda
 
 should_test_pickle
 
 check_gradgrad
 
 FIXME_no_cuda_gradgrad_comparison
 
 precision
 
- Public Attributes inherited from common_nn.TestBase
 desc
 
 fullname
 
 constructor
 
 reference_fn
 

Detailed Description

Definition at line 3146 of file common_nn.py.


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