U
    rh	                     @  sH   d dl mZ d dlZd dlZd dlmZ d dlmZ G dd deZ	dS )    )annotationsN)Base)expectc                   @  s   e Zd ZeddddZdS )DynamicQuantizeLinearNone)returnc                  C  sF  t jjddgdddgd} tddd	d
ddgtj}tdt|}t	dt
|}t|| d }ttd| | ddtj}tt|| | ddtj}t| |g|||gdd tdddd
ddgtj}tdt|}t	dt
|}t|| d }ttd| | ddtj}tt|| | ddtj}t| |g|||gdd tdddddddddddd gtjd!}tdt|}t	dt
|}t|| d }ttd| | ddtj}tt|| | ddtj}t| |g|||gd"d d S )#Nr   xyZy_scaleZy_zero_point)inputsoutputsr      g      gq=
ףp?g      ?   Ztest_dynamicquantizelinear)r
   r   nameg      g ggQ
g      Z'test_dynamicquantizelinear_max_adjusted   g @g?g      @gQ
@g      @g      ?g@g333333@g      @g(\@)      Z'test_dynamicquantizelinear_min_adjusted)onnxhelperZ	make_nodenparrayZastypeZfloat32ZminimumminmaximummaxZcliproundZuint8r   Zreshape)nodeXZx_minZx_maxZY_ScaleZY_ZeroPointY r   U/tmp/pip-unpacked-wheel-xnis5xre/onnx/backend/test/case/node/dynamicquantizelinear.pyexport   s\    "$"$""$zDynamicQuantizeLinear.exportN)__name__
__module____qualname__staticmethodr    r   r   r   r   r      s   r   )

__future__r   Znumpyr   r   Zonnx.backend.test.case.baser   Zonnx.backend.test.case.noder   r   r   r   r   r   <module>   s
   