U
    d                     @   s   d dl Z d dlmZ d dlmZ d dlZd dlm	Z	m
Z
 d dlm  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)core	workspacezNo MKLDNN support.c                   @   sJ   e Zd Zef eddeddedde dejdd Z	dS )	SoftmaxTest            )sizeinput_channels
batch_sizeinplacec           	      C   sL   t jddgdgdd}tj||||tjd }| |||gdg d S )NZSoftmaxXYr   )Zaxisg      ?r   )r   ZCreateOperatornprandomZrandZastypeZfloat32ZassertDeviceChecks)	selfr
   r   r   r   gcdcopr    r   G/tmp/pip-unpacked-wheel-ua33x9lu/caffe2/python/ideep/softmax_op_test.pytest_softmax   s"       zSoftmaxTest.test_softmaxN)
__name__
__module____qualname__r   stZintegersZbooleansmuZgcsr   r   r   r   r   r      s   

r   __main__)ZunittestZhypothesis.strategiesZ
strategiesr   Z
hypothesisr   Znumpyr   Zcaffe2.pythonr   r   Z"caffe2.python.hypothesis_test_utilpythonZhypothesis_test_utilhuZcaffe2.python.ideep_test_utilZideep_test_utilr   ZskipIfCZ
use_mkldnnZHypothesisTestCaser   r   mainr   r   r   r   <module>   s   