hypothesis_test_util Namespace Reference

Module caffe2.python.hypothesis_test_util. More...

Classes
class	HypothesisTestCase

Functions
def	is_sandcastle ()
def	dims (min_value=1, max_value=5)
def	elements_of_type (dtype=np.float32, filter_=None)
def	arrays (dims, dtype=np.float32, elements=None)
def	tensor (min_dim=1, max_dim=4, dtype=np.float32, elements=None, kwargs)
def	segment_ids (size, is_sorted)
def	lengths (size, kwargs)
def	segmented_tensor (min_dim=1, max_dim=4, dtype=np.float32, is_sorted=True, elements=None, segment_generator=segment_ids, allow_empty=False, kwargs)
def	lengths_tensor (args, kwargs)
def	sparse_segmented_tensor (min_dim=1, max_dim=4, dtype=np.float32, is_sorted=True, elements=None, allow_empty=False, segment_generator=segment_ids, kwargs)
def	sparse_lengths_tensor (kwargs)
def	tensors (n, min_dim=1, max_dim=4, dtype=np.float32, elements=None, kwargs)
def	device_checker_device_options ()
def	gradient_checker_device_option ()
def	temp_workspace (name=b"temp_ws")
def	runOpBenchmark (device_option, op, inputs, input_device_options=None, iterations=10)

Variables
	cpu_do = caffe2_pb2.DeviceOption()
	gpu_do = caffe2_pb2.DeviceOption(device_type=caffe2_pb2.CUDA)
list	device_options = [cpu_do] + ([gpu_do] if workspace.has_gpu_support else [])
list	expanded_device_options
	gcs
	gcs_cpu_only = dict(gc=st.sampled_from([cpu_do]), dc=st.just([cpu_do]))
	gcs_gpu_only = dict(gc=st.sampled_from([gpu_do]), dc=st.just([gpu_do]))

Detailed Description

Module caffe2.python.hypothesis_test_util.

Variable Documentation

expanded_device_options

list hypothesis_test_util.expanded_device_options

Initial value:

=  [cpu_do] + (
    [caffe2_pb2.DeviceOption(device_type=caffe2_pb2.CUDA, cuda_gpu_id=i)
     for i in range(workspace.NumCudaDevices())]
    if workspace.has_gpu_support else [])

Definition at line 213 of file hypothesis_test_util.py.

gcs

hypothesis_test_util.gcs

Initial value:

=  dict(
    gc=gradient_checker_device_option(),
    dc=device_checker_device_options()
)

Definition at line 227 of file hypothesis_test_util.py.