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

Public Member Functions

def __init__ (self, args, kwargs)
 
def constructor_args (self)
 
- Public Member Functions inherited from common_nn.ModuleTest
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

 cudnn
 
 check_inplace
 
 check_gradgrad
 
 skip_double
 
- Public Attributes inherited from common_nn.ModuleTest
 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 252 of file test_nn.py.


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