U
    d                     @   s   d Z ddlmZ ddlmZ ddlmZ ej	Z	ej
ejdZejeg Zdd Zdd	 Zee e d
Zeee	gee	gd
Zeeegeegd
Zeee	egee	egd
ZdS )z
The IDEEP test utils is a small addition on top of the hypothesis test utils
under caffe2/python, which allows one to more easily test IDEEP related
operators.
    N)
caffe2_pb2)hypothesis_test_util)Zdevice_typec                   C   s
   t tS N)stjustdevice_options r   r   A/tmp/pip-unpacked-wheel-ua33x9lu/caffe2/python/ideep_test_util.pydevice_checker_device_options   s    r
   c                   C   s
   t tS r   )r   sampled_fromr   r   r   r   r	   gradient_checker_device_option   s    r   )gcdc)__doc__Zhypothesis.strategiesZ
strategiesr   Zcaffe2.protor   Zcaffe2.pythonr   huZcpu_doZDeviceOptionZIDEEPZideep_dor   r
   r   dictZgcsr   r   Zgcs_cpu_onlyZgcs_ideep_onlyZgcs_cpu_ideepr   r   r   r	   <module>   s   