Caffe2 - Python API
A deep learning, cross platform ML framework
Public Member Functions | Static Public Attributes | List of all members
caffe2.python.serialized_test.serialized_test_util.SerializedTestCase Class Reference
Inheritance diagram for caffe2.python.serialized_test.serialized_test_util.SerializedTestCase:

Public Member Functions

def get_output_dir (self)
def get_output_filename (self)
def serialize_test (self, inputs, outputs, grad_ops, op, device_option)
def compare_test (self, inputs, outputs, grad_ops, atol=1e-7, rtol=1e-7)
def assertSerializedOperatorChecks (self, inputs, outputs, gradient_operator, op, device_option, atol=1e-7, rtol=1e-7)
def assertReferenceChecks (self, device_option, op, inputs, reference, input_device_options=None, threshold=1e-4, output_to_grad=None, grad_reference=None, atol=None, outputs_to_check=None)

Static Public Attributes

bool should_serialize = False

Detailed Description

Definition at line 65 of file

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