mkl_test_util Namespace Reference

Module caffe2.python.mkl_test_util. More...

Functions
def	device_checker_device_options ()
def	gradient_checker_device_option ()

Variables
	cpu_do = hu.cpu_do
	gpu_do = hu.gpu_do
	mkl_do = caffe2_pb2.DeviceOption(device_type=caffe2_pb2.MKLDNN)
	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]))
	gcs_mkl_only = dict(gc=st.sampled_from([mkl_do]), dc=st.just([mkl_do]))

Detailed Description

Module caffe2.python.mkl_test_util.

Variable Documentation

device_options

mkl_test_util.device_options

Initial value:

=  hu.device_options + (
    [mkl_do] if workspace.C.has_mkldnn else [])

Definition at line 23 of file mkl_test_util.py.

gcs

mkl_test_util.gcs

Initial value:

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

Definition at line 35 of file mkl_test_util.py.