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 serialized_test_util.py.

---

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

• caffe2/python/serialized_test/serialized_test_util.py