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                   @   sX   e Zd Zef eddeddeddedgdeje	dddd	 Z
d
S )LRNTest      
   ZNCHW)input_channels
batch_sizeim_sizeorderi'  )deadlinec           	   
   C   sh   t jddgddgdddd|d	}tj||||tj}| |||gd
g | |||gd
d
g d S )NZLRNXYZY_scale   gMbP?g      ?g       @)sizealphabetaZbiasr   r   )	r   ZCreateOperatornprandomZrandZastypeZfloat32ZassertDeviceChecksZassertGradientChecks)	selfr
   r   r   r   gcdcopr    r   C/tmp/pip-unpacked-wheel-ua33x9lu/caffe2/python/ideep/LRN_op_test.pytest_LRN   s(    	
   zLRNTest.test_LRNN)__name__
__module____qualname__r   stZintegersZsampled_from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_utilhuZcaffe2.python.ideep_test_utilZideep_test_utilr"   ZskipIfCZ
use_mkldnnZHypothesisTestCaser   r   mainr   r   r   r   <module>   s   