pytorch
1## @package ideep_test_util
2# Module caffe2.python.ideep_test_util
3"""
4The IDEEP test utils is a small addition on top of the hypothesis test utils
5under caffe2/python, which allows one to more easily test IDEEP related
6operators.
7"""
8
9
10
11
12
13
14import hypothesis.strategies as st15
16from caffe2.proto import caffe2_pb217from caffe2.python import hypothesis_test_util as hu18
19cpu_do = hu.cpu_do20ideep_do = caffe2_pb2.DeviceOption(device_type=caffe2_pb2.IDEEP)21device_options = hu.device_options + ([ideep_do])22
23
24def device_checker_device_options():25return st.just(device_options)26
27
28def gradient_checker_device_option():29return st.sampled_from(device_options)30
31
32gcs = dict(33gc=gradient_checker_device_option(),34dc=device_checker_device_options()35)
36
37gcs_cpu_only = dict(gc=st.sampled_from([cpu_do]), dc=st.just([cpu_do]))38gcs_ideep_only = dict(gc=st.sampled_from([ideep_do]), dc=st.just([ideep_do]))39gcs_cpu_ideep = dict(gc=st.sampled_from([cpu_do, ideep_do]), dc=st.just([cpu_do, ideep_do]))40