a
    ==icæ  ã                   @   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   ún/home/droni/.local/share/virtualenvs/DPS-5Je3_V2c/lib/python3.9/site-packages/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   )ÚgcZdc)Ú__doc__Zhypothesis.strategiesÚ
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   þ