U
    d                     @   s   d dl Z d dlmZ d dlmZmZ d dlZd dl	m
Z
mZ d dlm  mZ e ejj dG dd dejZedkre   dS )    N)givensettings)core	workspacezNo MKLDNN support.c                   @   sB   e Zd Zef ejejde	 dej
edddd ZdS )SigmoidTest)Zdtype)Xinplacei  )deadlinec                 C   sJ   t ddg|sdgndg}| |||gdg | |||gddg d S )NZSigmoidr   Yr   )r   ZCreateOperatorZassertDeviceChecksZassertGradientChecks)selfr   r   gcdcop r   G/tmp/pip-unpacked-wheel-ua33x9lu/caffe2/python/ideep/sigmoid_op_test.pytest_sigmoid   s    zSigmoidTest.test_sigmoidN)__name__
__module____qualname__r   huZtensornpZfloat32stZbooleansZgcsr   r   r   r   r   r   r      s   r   __main__)ZunittestZhypothesis.strategiesZ
strategiesr   Z
hypothesisr   r   Znumpyr   Zcaffe2.pythonr   r   Z"caffe2.python.hypothesis_test_utilpythonZhypothesis_test_utilr   ZskipIfCZ
use_mkldnnZHypothesisTestCaser   r   mainr   r   r   r   <module>   s   