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 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                   @   sH   e Zd Zef ejddejde	 de
jeddddd	 ZdS )
TransposeTest      )Zmin_dimZmax_dimZdtype)Xuse_axesN2   )deadlineZmax_examplesc                    s   t |j}t| tj  r>tjddgdg |d}ntjddgdg|d} fdd}| |||g| | 	|||gdg | 
|||gddg d S )	NZ	Transposer	   Y)axesdevice_option)r   c                    s$   rt j|  dgS t | gS d S )N)r   )npZ	transpose)r	   r   r
    I/tmp/pip-unpacked-wheel-ua33x9lu/caffe2/python/ideep/transpose_op_test.pytranspose_ref   s    z3TransposeTest.test_transpose.<locals>.transpose_refr   )lenshaper   Zarangerandomshuffler   ZCreateOperatorZassertReferenceChecksZassertDeviceChecksZassertGradientChecks)selfr	   r
   gcdcndimopr   r   r   r   test_transpose   s*    

       zTransposeTest.test_transpose)__name__
__module____qualname__r   huZtensorr   Zfloat32stZbooleansmuZ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caffe2.python.ideep_test_utilZideep_test_utilr$   ZskipIfCZ
use_mkldnnZHypothesisTestCaser   r   mainr   r   r   r   <module>   s   